move loss plot viz to separate script, and pass model path directly

3dlg-hcvc · Nov 2, 2022 · 4f209ba · 4f209ba
1 parent 4e3df95
commit 4f209ba
Show file tree

Hide file tree

Showing 4 changed files with 237 additions and 161 deletions.
diff --git a/README.md b/README.md
@@ -149,19 +149,19 @@ Next, we'll send the pairs of BEV texture maps to the model for scoring. You sho
 ```bash
 python scripts/test.py \
     --gpu_ids {COMMA SEPARATED GPU ID LIST} \
-    --model_results_dir {PATH TO FOLDER CONTAINING TRAINED MODEL} \
+    --model_ckpt_fpath {PATH TO TRAINED PYTORCH MODEL CHECKPOINT} \
     --config_name {FILE NAME OF YAML MODEL CONFIG} \
     --serialization_save_dir {PATH WHERE SERIALIZED PREDS WILL BE SAVED TO}
 ```
 Please note that the YAML config must be the name of a config under `salve/configs/*`.
 
 For example, if your model checkpoint was stored in a directory accessible at:
 ```
-/data/johnlam/models_for_lambert/2021_11_19_21_42_11
+/data/johnlambert/models/2021_11_19_21_42_11
 ```
 and contained pytorch model checkpoint (.pth) `train_ckpt.pth`, then we would use via CLI
 ```
---model_results_dir /data/johnlam/models_for_lambert/2021_11_19_21_42_11
+--model_ckpt_fpath /data/johnlambert/models/2021_11_19_21_42_11/train_ckpt.pth
 ```
 In the config you'll see a line:
 ```

diff --git a/salve/configs/9fcbb628bd5efffbdcc4ce55a9eb380d.yaml b/salve/configs/9fcbb628bd5efffbdcc4ce55a9eb380d.yaml
@@ -25,7 +25,8 @@ TrainingConfig:
     batch_size: 256
 
     #data_root: /srv/scratch/jlambert30/salve/ZinD_Bridge_API_BEV_2021_10_20_lowres
-    data_root: /srv/scratch/jlambert30/salve/2022_09_29_zind_texture_map_renderings_test
+    #data_root: /srv/scratch/jlambert30/salve/2022_09_29_zind_texture_map_renderings_test
+    data_root: /Users/johnlambert/Downloads/Renderings_ZinD_bridge_api_GT_WDO_2021_11_20_SE2_width_thresh0.8
     layout_data_root: 
     #model_save_dirpath": "/Users/johnlam/Downloads/ZinD_trained_models_2021_06_25",
     model_save_dirpath: /data/johnlam/ZinD_trained_models_2021_11_09