From d3b41e9a9684ae0453444423b277213eda043b98 Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sun, 8 May 2022 15:04:56 +0200 Subject: [PATCH] i dont know --- pytorch/pytorch.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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)