update seg readme

WangYueFt · Dec 10, 2018 · 35425de · 35425de
1 parent 04c6577
commit 35425de
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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
 ```