From fdfc07d3beb69a7e4fb95a5b51f399b22fe00c91 Mon Sep 17 00:00:00 2001 From: Franck Dary <franck.dary@lis-lab.fr> Date: Tue, 4 May 2021 17:18:35 +0200 Subject: [PATCH] Corrected path of train log file --- Train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Train.py b/Train.py index cd21874..a03d307 100644 --- a/Train.py +++ b/Train.py @@ -85,7 +85,7 @@ def evalModelAndSave(debug, model, ts, strat, dicts, modelDir, devFile, bestLoss devScore = ", Dev : UAS=%.2f"%(UAS) if saved : torch.save(model, modelDir+"/network.pt") - for out in [sys.stderr, open(modelDir+"train.log", "w" if epoch == 1 else "a")] : + for out in [sys.stderr, open(modelDir+"/train.log", "w" if epoch == 1 else "a")] : print("{} : Epoch {:{}}/{}, loss={:6.2f}{} {}".format(timeStamp(), epoch, len(str(nbIter)), nbIter, totalLoss, devScore, "SAVED" if saved else ""), file=out) return bestLoss, bestScore -- GitLab