aboutsummaryrefslogtreecommitdiff
path: root/xor.py
diff options
context:
space:
mode:
Diffstat (limited to 'xor.py')
-rw-r--r--xor.py2
1 files changed, 1 insertions, 1 deletions
diff --git a/xor.py b/xor.py
index f7e247f..905ddfb 100644
--- a/xor.py
+++ b/xor.py
@@ -144,7 +144,7 @@ def train_model(name: str):
loss = loss_fn(out, Y)
loss.backward()
optimizer.step()
- if (epoch+1) % 500 == 0:
+ if (epoch+1) % 100 == 0:
print(f" Epoch {epoch+1}, Loss: {loss.item():.4f}")
return net