From 35425de20ee6c2058466477e2ec5ce0685ae3557 Mon Sep 17 00:00:00 2001 From: Yongbin Sun Date: Mon, 10 Dec 2018 17:20:49 -0500 Subject: [PATCH] update seg readme --- part_seg/README.md | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/part_seg/README.md b/part_seg/README.md index 6b1c6db..959ee18 100644 --- a/part_seg/README.md +++ b/part_seg/README.md @@ -16,7 +16,7 @@ Train the model on 2 GPUs, each with 12 GB memeory. python train_multi_gpu.py ``` -Model parameters are saved every 10 epochs in "train_results/trained_models/". +Model parameters are saved every 5 epochs in "train_results/trained_models/". ### Evaluation @@ -26,8 +26,8 @@ To evaluate the model saved after epoch n, python test.py --model_path train_results/trained_models/epoch_n.ckpt ``` -For example, if we want to test the model saved after 160 epochs, +For example, if we want to test the model saved after 175 epochs (provided), ``` -python test.py --model_path train_results/trained_models/epoch_160.ckpt +python test.py --model_path train_results/trained_models/epoch_175.ckpt ```