File size: 976 Bytes
1f1004e | 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 | {
"train_mode": "traj",
"out_dir": "runs/pixcell_student_traj",
"feature_paths": "",
"feature_glob": "",
"manifest": "/common/users/wq50/UNI2_features/extracted_features/sae_manifests_tcga_patient_train_test_90_10.json",
"manifest_key": "train",
"feature_norm": "none",
"pix_model_id": "StonyBrook-CVLab/PixCell-256",
"pix_pipeline_id": "StonyBrook-CVLab/PixCell-pipeline",
"vae_model_id": "stabilityai/stable-diffusion-3.5-large",
"vae_subfolder": "vae",
"device": "cuda:5",
"dtype": "fp16",
"seed": 42,
"batch_size": 8,
"num_workers": 4,
"max_steps": 30000,
"lr": 1e-05,
"weight_decay": 0.0001,
"grad_clip": 1.0,
"latent_size": 32,
"teacher_steps": 28,
"traj_min_steps": 4,
"traj_max_steps": 8,
"guidance_teacher": 3.0,
"guidance_student": 1.0,
"init_student_from_teacher": true,
"ema": true,
"ema_decay": 0.999,
"log_every": 10,
"save_every": 1000,
"preview_every": 500,
"preview_steps": 6,
"resume": ""
} |