diff --git a/Train.py b/Train.py index cd218748eca9db025bb1862455a2ecc94e9e6bd3..a03d3079199ba3376f0a11b06f8a1e881a6f6698 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