From bf2f4b3a4c77df6452fdc4034db081a3ca0e1653 Mon Sep 17 00:00:00 2001 From: Max Burg <27483850+MaxFBurg@users.noreply.github.com> Date: Thu, 11 Apr 2024 17:33:15 +0200 Subject: [PATCH] update lr to 1 to match quickstart notebook Address Issue https://github.com/sinzlab/mei/issues/38 --- demo/dj_integration.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/demo/dj_integration.ipynb b/demo/dj_integration.ipynb index 686a6ce..5a7ef04 100644 --- a/demo/dj_integration.ipynb +++ b/demo/dj_integration.ipynb @@ -141,7 +141,7 @@ "method_fn = \"mei.methods.gradient_ascent\"\n", "method_config = dict(\n", " initial=dict(path=\"mei.initial.RandomNormal\"),\n", - " optimizer=dict(path=\"torch.optim.SGD\", kwargs=dict(lr=0.1)),\n", + " optimizer=dict(path=\"torch.optim.SGD\", kwargs=dict(lr=1)),\n", " stopper=dict(path=\"mei.stoppers.NumIterations\", kwargs=dict(num_iterations=1000)),\n", " objectives=[dict(path=\"mei.objectives.EvaluationObjective\", kwargs=dict(interval=10))],\n", " device=\"cuda\",\n",