Add missing param in initialize_optimizer (#245)

deepset-ai · Feb 13, 2020 · 5511899 · 5511899
1 parent 5e62b71
commit 5511899
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tutorials/1_farm_building_blocks.ipynb b/tutorials/1_farm_building_blocks.ipynb
@@ -257,6 +257,7 @@
     "\n",
     "model, optimizer, lr_schedule = initialize_optimizer(\n",
     "    model=model,\n",
+    "    device=device,\n",
     "    learning_rate=LEARNING_RATE,\n",
     "    n_batches=len(data_silo.loaders[\"train\"]),\n",
     "    n_epochs=N_EPOCHS)"