English
music
music-captioning
lp-music-caps / supervised.yaml
seungheondoh's picture
Upload 6 files
ba831b6
raw
history blame
444 Bytes
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