From a8192b0e3a47858d40066aede81276b72c7d1d90 Mon Sep 17 00:00:00 2001 From: JMGaljaard Date: Thu, 15 Sep 2022 21:17:26 +0200 Subject: [PATCH] Fix mistake in federated jinja template --- experiments/node.jinja.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/experiments/node.jinja.yaml b/experiments/node.jinja.yaml index 2950ee82..372d3f06 100644 --- a/experiments/node.jinja.yaml +++ b/experiments/node.jinja.yaml @@ -5,7 +5,7 @@ cuda: {{ task.get_learn_param('cuda') }} scheduler_step_size: {{ task.get_scheduler_param(tpe, 'scheduler_step_size') }} scheduler_gamma: {{ task.get_scheduler_param(tpe, 'scheduler_gamma') }} rounds: {{ task.get_learn_param('rounds') }} -epochs: {{ task.get_hyper_param(tpe, 'total_epochs') }} +epochs: {{ task.get_learn_param('epochs_per_round') }} lr: {{ task.get_optimizer_param(tpe, 'lr') }} momentum: {{ task.get_optimizer_param(tpe, 'momentum') }} shuffle: {{ task.get_sampler_param(tpe, 'shuffle') }}