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