.

anydoor/configs/anydoor.yaml

@@ -80,6 +80,4 @@ model:
 
     cond_stage_config:
       target: ldm.modules.encoders.modules.FrozenDinoV2Encoder
-      weight: /workspace/train-wefadoor-master/dinov2_vitg14_pretrain.pth
-    
-
+      weight: workspace/train-wefadoor-master/dinov2_vitg14_pretrain.pth

anydoor/configs/inference.yaml

@@ -1,3 +1,4 @@
-pretrained_model: /workspace/300k_wefa_boys_slim/lightning_logs/version_0/checkpoints/boys.ckpt
+
+pretrained_model: /workspace/mixed_wefa_unanon/step_300k_slim.ckpt
 config_file: configs/anydoor.yaml
 save_memory: False

anydoor/run_train_anydoor.py

@@ -42,7 +42,7 @@ if save_memory:
 # accumulate_grad_batches=1
 
 # Configs
-resume_path = '/workspace/train-wefadoor-master/anydoor/lightning_logs/version_5/checkpoints/step_300k.ckpt'
+resume_path = '/workspace/mixed_wefa_unanon/step_300k_slim.ckpt'
 batch_size = 8
 logger_freq = 1000
 learning_rate = 1e-5