diff --git a/config_collab.yaml b/config_collab.yaml new file mode 100644 index 0000000..de3d44e --- /dev/null +++ b/config_collab.yaml @@ -0,0 +1,60 @@ +# config.yaml +# --- Paths --- +cropped_raw_subdir: Cropped_Raw/ +cropped_raw_size: 2000 +secondary_align_csv: align_phase_v2_exposure_corr.csv +script_path: /content/drive/MyDrive/OnSight/traces +mlflow_path: /content/drive/MyDrive/OnSight/mlflow + +# --- Training Params --- +colorspace: lin_rec2020 +device: mps +batch_size: 2 +crop_size: 256 +lr_base: 2.5e-5 +clipping: 1e-2 +num_epochs_pretraining: 30 +num_epochs_finetuning: 20 +val_split: 0.2 +random_seed: 42 +cosine_annealing: False +iso_range: [0, 999999] + +# --- Experiment Settings --- +experiment: NAF_test +mlflow_experiment: NAFNet_variations + +# --- Run Configuration ---: +run_name: NAF_l1_SSIM +run_path: NAF_deep_test_align +model_params: + chans: [32, 64, 128, 256, 256, 256] + enc_blk_nums: [2, 2, 2, 3, 4] + middle_blk_num: 12 + dec_blk_nums: [2, 2, 2, 2, 2] + cond_input: 1 + in_channels: 4 + out_channels: 3 + rggb: True + use_CondFuserV2: False + use_add: False + use_CondFuserV3: False + use_attnblock: False + use_CondFuserV4: False + use_NAFBlock0_learned_norm: False + use_input_stats: False + gamma: 2.2 + +# --- Loss Configureation ---: +# alpha: 0.2 +# beta: 5.0 +# l1_weight: 0.16 +# ssim_weight: 0.84 +# tv_weight: 0.0 +# vgg_loss_weight: 0.0 +# percept_loss_weight: 0.025 + +dist_w: 0 +l1_w: 1 +msssim_w: 0 +alpha_beta_gamma: [1,1,1] \ No newline at end of file