From ed69bd98faf4a6e7257f76ad880a5f3a5ee780e9 Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sat, 7 May 2022 21:58:44 +0200 Subject: [PATCH] epochs run? --- pytorch/pytorch.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index 650d030..faddc97 100644 --- a/pytorch/pytorch.py +++ b/pytorch/pytorch.py @@ -38,6 +38,7 @@ ex.observers.append(FileStorageObserver('my_runs')) @ex.config def my_config(): epochs = 1500 + @@ -195,7 +196,7 @@ with open("result.txt", "w+") as file: @ex.main -def main(): +def main(epochs, _run): lr = 1e-6 my_config() print("number of epochs is: ", epochs)