Add replications to jinja templates to allow for setting correct logg…

…ing directory
JMGaljaard · Sep 6, 2022 · d43d1fc · d43d1fc
1 parent 30adc09
commit d43d1fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/experiments/dist_node.jinja.yaml b/experiments/dist_node.jinja.yaml
@@ -1,3 +1,4 @@
+replication: {{ replication }}
 batch_size: {{ task.get_hyper_param(tpe, 'bs') }}
 test_batch_size: {{ task.get_hyper_param(tpe, 'test_bs') }}
 cuda: {{ task.get_learn_param('cuda') }}
@@ -13,4 +14,3 @@ learning_rate: {{ task.get_optimizer_param(tpe, 'lr') }}
 learning_rate_decay: {{ task.get_hyper_param(tpe, 'lr_decay') }}
 seed: {{ task.get_net_param('seed') }}
 loss: {{ task.get_net_param('loss_function').value }}
-
diff --git a/experiments/node.jinja.yaml b/experiments/node.jinja.yaml
@@ -1,3 +1,4 @@
+replication: {{ replication }}
 batch_size: {{ task.get_hyper_param(tpe, 'bs') }}
 test_batch_size: {{ task.get_hyper_param(tpe, 'test_bs') }}
 cuda: {{ task.get_learn_param('cuda') }}