Fix Trainer init arguments

deepset-ai · Jan 17, 2020 · 9f11bc3 · 9f11bc3
1 parent 30999b3
commit 9f11bc3
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/farm/train.py b/farm/train.py
@@ -287,7 +287,13 @@ def _load_checkpoint(cls, path, data_silo):
         scheduler = get_scheduler(optimizer, scheduler_opts)
         scheduler.load_state_dict(scheduler_state_dict)
 
-        trainer = Trainer(data_silo=data_silo, model=model, lr_schedule=scheduler, **trainer_state_dict)
+        trainer = Trainer(
+            data_silo=data_silo,
+            model=model,
+            optimizer=optimizer,
+            lr_schedule=scheduler,
+            **trainer_state_dict
+        )
         return trainer
 
     def _save(self):