template.diff

diff --git a/tests/pytorch/nightly/wav2vec2.libsonnet b/tests/pytorch/nightly/wav2vec2.libsonnet
index 5c6cd8e..a3428ac 100644
--- a/tests/pytorch/nightly/wav2vec2.libsonnet
+++ b/tests/pytorch/nightly/wav2vec2.libsonnet
@@ -26,6 +26,8 @@ local utils = import 'templates/utils.libsonnet';
     export OMP_NUM_THREADS=1
     fairseq-hydra-train \
     task.data=/datasets/w2v2-librispeech-100hrs/w2v/manifest \
+    optimization.max_update %d \
+    dataset.batch_size 4 \
     --config-dir fairseq/examples/wav2vec/config/pretraining   \
     --config-name wav2vec2_large_librivox_tpu.yaml \
   |||,
@@ -40,15 +42,14 @@ local utils = import 'templates/utils.libsonnet';
   },
   local func = common.Functional {
     command: utils.scriptCommand(
-      |||
-        %(command_common)s  --max-update 500
-      ||| % command_common
+      command_common % 500
     ),
   },
   local conv = common.Convergence {
     command: utils.scriptCommand(
+      (command_common % 50000) + '\'
+        +
       |||
-        %(command_common)s  --max-update 50000 \
           2>&1 | tee training_logs.txt
         loss=$(
           cat training_logs.txt | grep '| loss ' | \
@@ -56,7 +57,7 @@ local utils = import 'templates/utils.libsonnet';
         )
         echo 'final loss is' $loss
         test $( echo $loss | cut -d '.' -f1 ) -lt 3
-      ||| % command_common
+      |||
     ),
   },
   local v3_8 = {