anonymous
adding models
39089d7
raw
history blame contribute delete
No virus
979 Bytes
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