add 1x1 config for transformer xl

reiase · Sep 26, 2023 · 95b2b44 · 95b2b44
1 parent 36e18b0
commit 95b2b44
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/training/nvidia/transformer_xl-pytorch/config/config_A100x1x1.py b/training/nvidia/transformer_xl-pytorch/config/config_A100x1x1.py
@@ -0,0 +1,6 @@
+from config_common import *
+
+train_batch_size = 60
+eval_batch_size = 60
+
+seed = 42