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

update dllib-mlflow.py
Some checks failed
s444356-evaluation/pipeline/head This commit looks good
Details
s444356-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Maciej Czajka 2022-05-11 15:21:21 +02:00
parent e10b25c9c9
commit 09d0026265
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
Biblioteka_DL/dllib-mlflow.py
View File

@ -291,7 +291,8 @@ def my_main(epochs):
loss_fn = nn.CrossEntropyLoss() loss_fn = nn.CrossEntropyLoss()
# epochs = 1000 # epochs = 1000
# epochs = epochs # epochs = epochs
mlflow.log_param("epochs", epochs)
def print_(loss): def print_(loss):
print ("The loss calculated: ", loss) print ("The loss calculated: ", loss)
@ -304,6 +305,8 @@ def my_main(epochs):
loss = loss_fn(y_pred, y_train.squeeze(-1)) loss = loss_fn(y_pred, y_train.squeeze(-1))
print_(loss.item()) print_(loss.item())
mlflow.log_param("loss", loss.item)
# Zero gradients # Zero gradients
optimizer.zero_grad() optimizer.zero_grad()
loss.backward() # Gradients loss.backward() # Gradients
@ -328,5 +331,4 @@ def my_main(epochs):
with mlflow.start_run() as run: with mlflow.start_run() as run:
my_main(epochs) my_main(epochs)
mlflow.log_param("epochs", epochs)