Spaces:
Runtime error
Runtime error
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 | |