Fix fine tuning notebook (#268)

elixir-nx · Oct 25, 2023 · 2f35d4e · 2f35d4e
1 parent 38876bb
commit 2f35d4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/notebooks/fine_tuning.livemd b/notebooks/fine_tuning.livemd
@@ -276,7 +276,7 @@ loss =
     sparse: true
   )
 
-optimizer = Polaris.Optimizers.adam(5.0e-5)
+optimizer = Polaris.Optimizers.adam(learning_rate: 5.0e-5)
 
 loop = Axon.Loop.trainer(logits_model, loss, optimizer, log: 1)
 ```