diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py
index 96099e1..91ac241 100644
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@@ -154,7 +154,8 @@ def evaluate(model, val_loader):
     return model.validation_epoch_end(outputs)
     
 @ex.capture
-def fit(epochs, lr, model, train_loader, val_loader, _run, opt_func=torch.optim.SGD):
+def fit(epochs, lr, model, train_loader, val_loader, _log, opt_func=torch.optim.SGD):
+    _log.info("log info test ")
     epochs=epochs
     history = []
     optimizer = opt_func(model.parameters(), lr)