athenlab/mai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix batch loss reporting

Browse Source
This commit is contained in:
Kenneth
2024-05-19 00:07:34 +01:00
parent 5d7960d4c7
commit c8d8ec9928
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
train.py
View File

@@ -173,8 +173,8 @@ def train():
accuracy = 100 * correct / sample_size accuracy = 100 * correct / sample_size
running_loss += loss.item() running_loss += loss.item()
if i % 1000 == 999: if i % 10 == 9:
last_loss = running_loss / 1000 # loss per batch last_loss = running_loss / 10 # loss per batch
print(" batch {} loss: {}".format(i + 1, last_loss)) print(" batch {} loss: {}".format(i + 1, last_loss))
running_loss = 0.0 running_loss = 0.0