diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 71b4574..b4a15ea 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -206,6 +206,6 @@ def main():
     lr = 1e-6
     #my_config()
     #print("number of epochs is: ", epochs)
-    history5 = fit()
+    history5 = fit(lr)
 
 #ex.run()