From ccbcdd3715b7f87bfaa539242a03bd86af088311 Mon Sep 17 00:00:00 2001 From: Kornelia Girejko Date: Sun, 8 May 2022 20:30:23 +0200 Subject: [PATCH] Zaktualizuj 'train-sacred.py' --- train-sacred.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/train-sacred.py b/train-sacred.py index f09da5c..677305f 100644 --- a/train-sacred.py +++ b/train-sacred.py @@ -84,17 +84,17 @@ def my_main(epochs): loss.backward() optimizer.step() optimizer.zero_grad() - - if (epochs%100==0): - print(f'epoch:{epochs+1},loss = {loss.item():.4f}') + with open ("output.txt",'a+') as f: + if (epochs%100==0): + f.write(f'epoch:{epochs+1},loss = {loss.item():.4f}') with torch.no_grad(): y_predicted = model(X_testing) y_predicted_cls = y_predicted.round() acc = y_predicted_cls.eq(y_testing).sum()/float(y_testing.shape[0]) print(f'{acc:.4f}') - result = open("output",'w+') - result.write(f'{y_predicted}') + #result = open("output",'w+') + #result.write(f'{y_predicted}') torch.save(model, "modelS.pkl")