training_args_full.json

{
  "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": ""
}