File size: 480 Bytes
705a8fd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | batch_size: 16
context_size: 4
datasets:
avw_4k:
data_folder: /path/to/dataset/avw_4k
goals_per_obs: 4
test: /path/to/data_splits/avw_4k/val
train: /path/to/data_splits/avw_4k/train
distance:
max_dist_cat: 16
min_dist_cat: -16
from_checkpoint: /path/to/pretrained/cdit_b_100000.pth.tar
grad_clip_val: 10.0
image_size: 224
len_traj_pred: 16
lr: 16.0e-05
model: AVCDiT-B/2
normalize: true
num_workers: 1
results_dir: logs
run_name: training_stage1
train: true
|