Added launch target for transformerxl training.

This commit is contained in:
Debadeepta Dey 2021-08-12 02:27:12 +00:00 коммит произвёл Gustavo Rosa
Родитель be4a592e70
Коммит 917daeed56
1 изменённых файлов: 8 добавлений и 0 удалений

8
.vscode/launch.json поставляемый
Просмотреть файл

@ -267,6 +267,14 @@
"args": ["--results-dir", "C:\\Users\\dedey\\Documents\\archaiphilly\\phillytools\\bilevel_default_20200521",
"--out-dir", "C:\\Users\\dedey\\archai_experiment_reports", "--collate"]
},
{
"name": "Train Transformer XL",
"type": "python",
"request": "launch",
"program": "${cwd}/archai/nlp/nvidia_transformer_xl/train.py",
"console": "integratedTerminal",
"args": ["--config", "dgx1_1gpu_fp16", "--config_file", "wt103_base.yaml", "--attn_type", "2", "--n_layer", "16", "--n_head", "8", "--d_model", "256", "--d_head", "32", "--d_inner", "768", "--log_interval", "10"]
},
{
"name": "CurrentFile",
"type": "python",