latest run

anydoor/.gitattributes

@@ -0,0 +1 @@
+wefa_init.ckpt filter=lfs diff=lfs merge=lfs -text

anydoor/configs/datasets.yaml

@@ -32,7 +32,7 @@ Train:
     image_dir: /mnt/workspace/xizhi/data/MVImgNet/
 
   VitonHD:
-    image_dir: /workspace/current_wefa/cloth
+    image_dir: /workspace/mixed_dataset/cloth
 
   Dresscode:
     image_dir: /mnt/workspace/xizhi/data/dresscode/DressCode/upper_body/label_maps/
@@ -62,7 +62,7 @@ Test:
     bg_dir: path/DreamBooth/v1_800
 
   VitonHDTest:
-    image_dir: /workspace/current_wefa/cloth
+    image_dir: /workspace/mixed_dataset/cloth
 
 
 

anydoor/configs/inference.yaml

@@ -1,3 +1,3 @@
-pretrained_model: /workspace/200k_wefa_boys/lightning_logs/version_1/checkpoints/200k_slim.ckpt
+pretrained_model: /workspace/300k_wefa_boys_slim/lightning_logs/version_0/checkpoints/boys.ckpt
 config_file: configs/anydoor.yaml
 save_memory: False

anydoor/run_inference_api_select.py

@@ -32,7 +32,7 @@ def load_model(new_model_ckpt):
     global model, ddim_sampler, current_model_ckpt
     if new_model_ckpt != current_model_ckpt:
         print(f"Loading new model: {new_model_ckpt}")
-        model.load_state_dict(load_state_dict(f'/workspace/200k_wefa_boys/lightning_logs/version_1/checkpoints/{new_model_ckpt}', location='cuda'))
+        model.load_state_dict(load_state_dict(f'/workspace/300k_wefa_boys_slim/lightning_logs/version_0/checkpoints/{new_model_ckpt}', location='cuda'))
         current_model_ckpt = new_model_ckpt
         print("New model loaded successfully.")
     else:

anydoor/run_train_anydoor.py

@@ -31,11 +31,21 @@ disable_verbosity()
 if save_memory:
     enable_sliced_attention()
  
+# # Config slower for less images
+# resume_path = '/workspace/wefa_init.ckpt'
+# batch_size = 8
+# logger_freq = 1000
+# learning_rate = 1e-7
+# sd_locked = False
+# only_mid_control = False
+# n_gpus = 2
+# accumulate_grad_batches=1
+
 # Configs
-resume_path = '/workspace/200k_wefa_boys/lightning_logs/version_1/checkpoints/200k_slim.ckpt'
+resume_path = '/workspace/train-wefadoor-master/anydoor/lightning_logs/version_5/checkpoints/step_300k.ckpt'
 batch_size = 8
 logger_freq = 1000
-learning_rate = 1e-7
+learning_rate = 1e-5
 sd_locked = False
 only_mid_control = False
 n_gpus = 2