From 551189933f9ea65f5acda4f4f5fed1db1e890c6a Mon Sep 17 00:00:00 2001
From: Tanay Soni <tanaysoni12@gmail.com>
Date: Thu, 13 Feb 2020 14:41:21 +0100
Subject: [PATCH] Add missing param in initialize_optimizer (#245)

---
 tutorials/1_farm_building_blocks.ipynb | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tutorials/1_farm_building_blocks.ipynb b/tutorials/1_farm_building_blocks.ipynb
index 87241d59d..db9df123f 100644
--- 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)"