GRADE / src /models /ours_diffusion /config.yaml
Bin-0815's picture
Release all GRADE models, checkpoints, and reviewed evaluation code (part 2)
f348660 verified
Raw History Blame Contribute Delete
464 Bytes
training: {batch_size: 1, mixed_precision: fp16, seed: 42}
data:
smoke_eval_root: ../../../evaluation_dataset/Smoke-Eval
resolution: {height: 288, width: 512}
scale_factor: 0.001
max_depth_m: 11.2
num_frames: 1
num_workers: 0
pretrained:
radar_model: ../../../checkpoints/grade/radar.safetensors
unet: ../../../checkpoints/grade/diffusion.safetensors
diffusion: {num_train_timesteps: 1000}
inference: {frame_skip: 1, batch_size: 1, num_workers: 0}