| {"embedder_config": "weights/WAM/configs/embedder.yaml", "augmentation_config": "weights/WAM/configs/all_augs_multi_wm.yaml", "extractor_config": "weights/WAM/configs/extractor.yaml", "attenuation_config": "weights/WAM/configs/attenuation.yaml", "embedder_model": "vae_small", "extractor_model": "sam_base", "nbits": 32, "img_size": 256, "img_size_extractor": 256, "attenuation": "jnd_1_3_blue", "scaling_w": 2.0, "scaling_w_schedule": null, "scaling_i": 1.0, "roll_probability": 0.2, "multiple_w": 1.0, "nb_wm_eval": 5, "optimizer": "AdamW,lr=1e-4", "optimizer_d": null, "scheduler": "CosineLRScheduler,lr_min=1e-6,t_initial=100,warmup_lr_init=1e-6,warmup_t=5", "epochs": 200, "batch_size": 8, "batch_size_eval": 16, "temperature": 1.0, "workers": 8, "to_freeze_embedder": null, "lambda_w": 1.0, "lambda_w2": 6.0, "lambda_i": 0.0, "lambda_d": 0.0, "balanced": true, "total_gnorm": 0.0, "perceptual_loss": "none", "disc_start": 0, "disc_num_layers": 2, "only_eval": false, "eval_freq": 5, "saveimg_freq": 5, "saveckpt_freq": 50, "seed": 42, "debug_slurm": false, "local_rank": 0, "master_port": 18482, "is_slurm_job": true, "n_nodes": 1, "node_id": 0, "global_rank": 0, "world_size": 8, "n_gpu_per_node": 8, "is_master": true, "multi_node": false, "distributed": true} | |