Streamer272/soc-2024
0
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🐛 Fixes test loss value

Browse Source
This commit is contained in:
Daniel Svitan 2025-01-06 20:12:40 +01:00
parent f6eafc28ec
commit 2f3c547b55
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
train_nn.py
View File

@ -119,7 +119,7 @@ for epoch in range(epochs):
pred = model(X) pred = model(X)
loss = loss_fn(pred, y) loss = loss_fn(pred, y)
test_loss = loss.item() * X.size(0) test_loss += loss.item() * X.size(0)
test_loss /= len(test_dataset) test_loss /= len(test_dataset)
test_losses.append(test_loss) test_losses.append(test_loss)