pretrain check

yswi · yswi · commit 237352bf6368 · 2022-10-10T11:36:19.000-04:00
diff --git a/README.md b/README.md
@@ -38,4 +38,9 @@ Alternatively, you can manually download the datasets and pretrained models from
 ## Step 2: Pretrain nominal shapes
 ```
 python pretrain.py --name <log name> --gpu_id 0
-```
+```
+If you want to check the result of your pretrained model, 
+```
+python pretrain.py --checkpoints_dir <dir> --gpu_id 0
+```
+then you will see the nominal reconstructions in /output/ directory.
