kuhyx/WUT_Computer_Science
1
0
Fork 0
mirror of https://github.com/kuhyx/WUT_Computer_Science.git synced 2026-07-04 18:23:15 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat: print info about epoch accuracy

Browse Source
This commit is contained in:
Krzysztof Rudnicki 2023-05-16 23:37:16 +02:00
parent 8d40eb6078
commit 51eab8bf82
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lab5/code/main.py
View File

@ -105,14 +105,14 @@ def single_train_iteration(
loss.backward()
training_parameters['optimizer'].step()
# Print loss value for every learning step
"""if (batch_idx + 1) % 100 == 0:
if (batch_idx + 1) % 100 == 0:
print(
f'''
Epoch [{epoch+1}/{training_parameters['hyperparameters']["num_epochs"]}],
Step [{batch_idx+1}/{len(training_parameters['loaders']['train_loader'])}],
Loss: {loss.item():.4f}
'''
)"""
)
# Append loss value for every learning step
loss_values.append(loss.item())
return data, training_parameters['optimizer']