s444356/ium_444356
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

update dllib-mlflow.py
All checks were successful
s444356-evaluation/pipeline/head This commit looks good
Details
s444356-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Maciej Czajka 2022-05-09 22:14:03 +02:00
parent 8f297e82de
commit 5f9785d4fd
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Biblioteka_DL/dllib-mlflow.py
View File

@ -289,7 +289,7 @@ def my_main(epochs):
loss_fn = nn.CrossEntropyLoss() loss_fn = nn.CrossEntropyLoss()
# epochs = 1000 # epochs = 1000
# epochs = epochs # epochs = epochs
alpha = int(sys.argv[1]) if len(sys.argv) > 1 else 20 epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20
mlflow.log_param("epochs", epochs) mlflow.log_param("epochs", epochs)
def print_(loss): def print_(loss):