зеркало из https://github.com/microsoft/archai.git
Added launch target for transformerxl training.
This commit is contained in:
Родитель
be4a592e70
Коммит
917daeed56
|
@ -267,6 +267,14 @@
|
||||||
"args": ["--results-dir", "C:\\Users\\dedey\\Documents\\archaiphilly\\phillytools\\bilevel_default_20200521",
|
"args": ["--results-dir", "C:\\Users\\dedey\\Documents\\archaiphilly\\phillytools\\bilevel_default_20200521",
|
||||||
"--out-dir", "C:\\Users\\dedey\\archai_experiment_reports", "--collate"]
|
"--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",
|
"name": "CurrentFile",
|
||||||
"type": "python",
|
"type": "python",
|
||||||
|
|
Загрузка…
Ссылка в новой задаче