framework: bart | |
data_dir: ../../dataset | |
train_data: music_caps | |
text_type: gt | |
arch: transformer | |
workers: 8 | |
epochs: 100 | |
warmup_epochs: 1 | |
start_epoch: 0 | |
batch_size: 64 | |
world_size: 1 | |
lr: 0.0001 | |
min_lr: 1.0e-09 | |
rank: 0 | |
dist_url: tcp://localhost:12312 | |
dist_backend: nccl | |
seed: null | |
gpu: 0 | |
print_freq: 100 | |
multiprocessing_distributed: false | |
cos: true | |
bart_pretrain: false | |
label_smoothing: 0.1 | |
use_early_stopping: false | |
eval_sample: 64 | |
max_length: 128 | |