Spaces:
Runtime error
Runtime error
File size: 979 Bytes
39089d7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 |
debug: False mixed_precision: False resume_path: None experiment: global_finetune dataset: JSRT img_size: 128 data_dir: /vol/biodata/data/chest_xray/JSRT num_workers: 4 dim: 64 dim_mults: (1, 2, 4, 8) seg_out_dim: 1 img_out_dim: 4 img_inter_dim: 32 timesteps: 1000 beta_schedule: cosine objective: pred_noise tau: 0.1 global_model_path: logs/global_cl/None/20230627_183826/best_model.pt glob_loc_model_path: None unfreeze_weights_at_step: 0 augment_at_finetuning: False batch_size: 16 lr: 0.0001 weight_decay: 0 max_steps: 1000 p2_loss_weight_gamma: 0.0 p2_loss_weight_k: 1.0 device: cuda seed: 0 log_freq: 50 val_freq: 50 val_steps: 250 log_dir: /vol/biomedic3/mmr12/projects/TEDM/logs/global_finetune/98/20230628_154833 n_sampled_imgs: 8 max_val_steps: -1 saved_diffusion_model: logs/20230127_164150/best_model.pt t_steps_to_save: [50, 200, 400, 600, 800] n_labelled_images: 98 shared_weights_over_timesteps: False early_stop: False normalize: True channels: 1 out_channels: 1 |