GRADE / config.yaml
Bin-0815's picture
Release GRADE full model and checkpoints
4401ce6 verified
Raw History Blame Contribute Delete
488 Bytes
training: {batch_size: 1, mixed_precision: fp16, seed: 42}
data:
smoke_eval_root: inputs
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
controlnet: checkpoints/grade/control.safetensors
diffusion: {num_train_timesteps: 1000}
inference: {frame_skip: 1, batch_size: 1, num_workers: 0, output_root: outputs}