Fixed default wandb project name

ohayonguy · Oct 11, 2024 · c344cec · c344cec
1 parent 599f7fb
commit c344cec
Show file tree

Hide file tree

Showing 22 changed files with 22 additions and 22 deletions.
diff --git a/scripts/train/blind_face_restoration/train_pmrf.sh b/scripts/train/blind_face_restoration/train_pmrf.sh
@@ -21,7 +21,7 @@ python train.py \
 --weight_decay 1e-2 \
 --lr 5e-4 \
 --source_noise_std 0.1 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Blind face restoration PMRF" \
 --mmse_model_arch "swinir_L" \
 --mmse_model_ckpt_path "./checkpoints/swinir_restoration512_L1.pth"  # Path to the DifFace checkpoint of SwinIR
diff --git a/scripts/train/controlled_experiments/colorization/train_mmse.sh b/scripts/train/controlled_experiments/colorization/train_mmse.sh
@@ -20,5 +20,5 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior mean predictor"
diff --git a/scripts/train/controlled_experiments/colorization/train_naive_flow.sh b/scripts/train/controlled_experiments/colorization/train_naive_flow.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Naive Flow" \
 --source_noise_std 0.1
diff --git a/scripts/train/controlled_experiments/colorization/train_pmrf.sh b/scripts/train/controlled_experiments/colorization/train_pmrf.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "PMRF" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/...ts/train/controlled_experiments/colorization/train_posterior_conditioned_on_mmse_model.sh b/...ts/train/controlled_experiments/colorization/train_posterior_conditioned_on_mmse_model.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on the posterior mean" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/colorization/train_posterior_conditioned_on_y.sh b/scripts/train/controlled_experiments/colorization/train_posterior_conditioned_on_y.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on Y" \
 --conditional
diff --git a/scripts/train/controlled_experiments/denoising/train_mmse.sh b/scripts/train/controlled_experiments/denoising/train_mmse.sh
@@ -20,5 +20,5 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior mean predictor"
diff --git a/scripts/train/controlled_experiments/denoising/train_naive_flow.sh b/scripts/train/controlled_experiments/denoising/train_naive_flow.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Naive Flow" \
 --source_noise_std 0.025
diff --git a/scripts/train/controlled_experiments/denoising/train_pmrf.sh b/scripts/train/controlled_experiments/denoising/train_pmrf.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "PMRF" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/denoising/train_posterior_conditioned_on_mmse_model.sh b/scripts/train/controlled_experiments/denoising/train_posterior_conditioned_on_mmse_model.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on the posterior mean" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/denoising/train_posterior_conditioned_on_y.sh b/scripts/train/controlled_experiments/denoising/train_posterior_conditioned_on_y.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on Y" \
 --conditional
diff --git a/scripts/train/controlled_experiments/inpainting/train_mmse.sh b/scripts/train/controlled_experiments/inpainting/train_mmse.sh
@@ -20,5 +20,5 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior mean predictor"
diff --git a/scripts/train/controlled_experiments/inpainting/train_naive_flow.sh b/scripts/train/controlled_experiments/inpainting/train_naive_flow.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Naive Flow" \
 --source_noise_std 0.1
diff --git a/scripts/train/controlled_experiments/inpainting/train_pmrf.sh b/scripts/train/controlled_experiments/inpainting/train_pmrf.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "PMRF" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/inpainting/train_posterior_conditioned_on_mmse_model.sh b/scripts/train/controlled_experiments/inpainting/train_posterior_conditioned_on_mmse_model.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on the posterior mean" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/inpainting/train_posterior_conditioned_on_y.sh b/scripts/train/controlled_experiments/inpainting/train_posterior_conditioned_on_y.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on Y" \
 --conditional
diff --git a/scripts/train/controlled_experiments/super_resolution/train_mmse.sh b/scripts/train/controlled_experiments/super_resolution/train_mmse.sh
@@ -20,5 +20,5 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior mean predictor"
diff --git a/scripts/train/controlled_experiments/super_resolution/train_naive_flow.sh b/scripts/train/controlled_experiments/super_resolution/train_naive_flow.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Naive Flow" \
 --source_noise_std 0.1
diff --git a/scripts/train/controlled_experiments/super_resolution/train_pmrf.sh b/scripts/train/controlled_experiments/super_resolution/train_pmrf.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "PMRF" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/...rain/controlled_experiments/super_resolution/train_posterior_conditioned_on_mmse_model.sh b/...rain/controlled_experiments/super_resolution/train_posterior_conditioned_on_mmse_model.sh
@@ -20,7 +20,7 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on the posterior mean" \
 --mmse_model_arch "swinir_M" \
 --mmse_model_ckpt_path "/path/to/checkpoint/last.ckpt" \

diff --git a/scripts/train/controlled_experiments/super_resolution/train_posterior_conditioned_on_y.sh b/scripts/train/controlled_experiments/super_resolution/train_posterior_conditioned_on_y.sh
@@ -20,6 +20,6 @@ python train.py \
 --t_schedule "stratified_uniform" \
 --weight_decay 1e-2 \
 --lr 5e-4 \
---wandb_project_name "Posterior-Mean Rectified Flow" \
+--wandb_project_name "PMRF" \
 --wandb_group "Posterior conditioned on Y" \
 --conditional
diff --git a/train.py b/train.py
@@ -158,7 +158,7 @@ def main(args):
                         help='Optimizer learning rate.')
     parser.add_argument('--betas', type=tuple, required=False, default=(0.9, 0.95),
                         help='Betas for the AdamW optimizer.')
-    parser.add_argument('--wandb_project_name', type=str, required=True, default='Rectified Restoration Flow',
+    parser.add_argument('--wandb_project_name', type=str, required=True, default='PMRF',
                         help='Project name for weights and biases logger.')
     parser.add_argument('--wandb_group', type=str, required=False, default=None,
                         help='Group of wandb experiment.')