Zaktualizuj 'train-sacred.py'

2022-05-08 19:07:49 +02:00 · 2022-05-08 19:07:49 +02:00 · da3b0bbee8
commit da3b0bbee8
parent 806dff6116
1 changed files with 1 additions and 3 deletions
--- a/train-sacred.py
+++ b/train-sacred.py
@ -75,7 +75,7 @@ criterion = nn.BCELoss()
 optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

@ex.automain
-def my_main(epochs, _run):
+def my_main(epochs):
 # Trening
 #num_epochs = EPOCHS
    for epochs in range(epochs):
@ -98,7 +98,5 @@ def my_main(epochs, _run):

        torch.save(model, "modelP.pkl")
        
-        _run.info['accuracy'] = accuracy_score(y_testing, np.argmax(y_predicted, axis=1))
-        
        
 ex.run()