File size: 250 Bytes
9a964a6 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 | log_dir: "Checkpoint"
save_freq: 10
device: "cuda"
epochs: 100
batch_size: 32
pretrained_model: ""
train_data: "Data/train_list.txt"
val_data: "Data/val_list.txt"
num_workers: 16
optimizer_params:
lr: 0.0003
loss_params:
lambda_f0: 0.1
|