| FAIRSEQ= # Setup your fairseq directory | |
| config_dir=${FAIRSEQ}/examples/mr_hubert/config | |
| config_name=mr_hubert_base_librispeech | |
| # Prepared Data Directory | |
| data_dir=librispeech | |
| # -- data_dir | |
| # -- train.tsv | |
| # -- valid.tsv | |
| label_dir=labels | |
| # -- label_dir | |
| # -- train.km | |
| # -- valid.km | |
| # -- dict.km.txt | |
| exp_dir=exp # Target experiments directory | |
| ratios="[1, 2]" # Default label rate ratios | |
| label_rate=50 # Base label rate | |
| _opts= | |
| # If use slurm, uncomment this line and modify the job submission at | |
| # _opts="${_opts} hydra/launcher=submitit_slurm +hydra.launcher.partition=${your_slurm_partition} +run=submitit_reg" | |
| # If want to set additional experiment tag, uncomment this line | |
| # _opts="${_opts} hydra.sweep.subdir=${your_experiment_tag}" | |
| python ${FAIRSEQ}/fairseq_cli/hydra_train.py \ | |
| -m --config-dir ${config_dir} --config-name ${config_name} ${_opts} \ | |
| task.data=${data_dir} \ | |
| task.label_dir=${label_dir} \ | |
| task.labels='["km"]' \ | |
| model.label_rate=${label_rate} \ | |
| task.label_rate_ratios='${ratios}' \ | |
| hydra.sweep.dir=${exp_dir} & | |