fixes for default argument for vae decoder path

deepvk · Apr 22, 2024 · 94ca232 · 94ca232
1 parent f6c73f0
commit 94ca232
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scripts/metr_all_att/fid_gt_all_att_metr.sh b/scripts/metr_all_att/fid_gt_all_att_metr.sh
@@ -27,7 +27,7 @@ for ((i=0; i<${#attacks[@]}; i++)); do
       --run_generation \
       --additional_metrics \
       --run_no_w \
-      --image_folder /data/varlamov_a_data/tree-ring-watermark/msg_fid_gt_all_att_no_vae/${names[i]}  \
+      --image_folder msg_fid_gt_all_att_no_vae/${names[i]}  \
       --msg_type binary \
       --use_random_msgs \
       --msg_scaler 100 \
@@ -43,7 +43,7 @@ accelerate launch -m metr.run_metr_fid\
   --run_generation \
   --additional_metrics \
   --run_no_w \
-  --image_folder /data/varlamov_a_data/tree-ring-watermark/msg_fid_gt_all_att_no_vae/diff_150  \
+  --image_folder msg_fid_gt_all_att_no_vae/diff_150  \
   --msg_type binary \
   --use_random_msgs \
   --msg_scaler 100 \
@@ -60,7 +60,7 @@ accelerate launch -m metr.run_metr_fid\
   --run_generation \
   --additional_metrics \
   --run_no_w \
-  --image_folder /data/varlamov_a_data/tree-ring-watermark/msg_fid_gt_all_att_no_vae/vae_2018_q_1  \
+  --image_folder msg_fid_gt_all_att_no_vae/vae_2018_q_1  \
   --msg_type binary \
   --use_random_msgs \
   --msg_scaler 100 \
@@ -78,7 +78,7 @@ accelerate launch -m metr.run_metr_fid\
   --run_generation \
   --additional_metrics \
   --run_no_w \
-  --image_folder /data/varlamov_a_data/tree-ring-watermark/msg_fid_gt_all_att_no_vae/no_attack3  \
+  --image_folder msg_fid_gt_all_att_no_vae/no_attack3  \
   --msg_type binary \
   --use_random_msgs \
   --msg_scaler 100 
diff --git a/src/metr/run_metr.py b/src/metr/run_metr.py
@@ -346,7 +346,7 @@ def main(args):
     parser.add_argument('--vae_attack_quality', default=3, type=int)
 
     # METR++
-    parser.add_argument('--decoder_state_dict_path', default='sd2_decoder.pth')
+    parser.add_argument('--decoder_state_dict_path', default='finetune_ldm_decoder/ldm_decoder_checkpoint_000.pth')
     parser.add_argument('--no_stable_sig', action='store_true')
     parser.add_argument('--stable_sig_full_model_config', default="v2-inference.yaml")
     parser.add_argument('--stable_sig_full_model_ckpt', default='v2-1_512-ema-pruned.ckpt')