Zaktualizuj 'biblioteka_DL'

2022-05-06 19:21:59 +02:00 · 2022-05-06 19:21:59 +02:00 · aff6e0827b
commit aff6e0827b
parent a1429eea6e
1 changed files with 3 additions and 1 deletions
--- a/2
+++ b/2
@ -74,3 +74,5 @@ with torch.no_grad():
    print(f'{acc:.4f}')
    result = open("output",'w+')
    result.write(f'{y_predicted}')
+
+torch.save(model, "model.pkl")