From 254da57d4b1c0f43f1089cdc19ac87464dfe7c40 Mon Sep 17 00:00:00 2001
From: piotr6789 <piotr.ulanicki4@gmail.com>
Date: Sun, 25 Apr 2021 23:49:57 +0200
Subject: [PATCH] add pytorch exercise

---
 pytorch-example.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pytorch-example.py b/pytorch-example.py
index afd6f46..75c8ab7 100644
--- a/pytorch-example.py
+++ b/pytorch-example.py
@@ -48,7 +48,7 @@ criterion = torch.nn.BCELoss(reduction='mean')
 optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate)
 
 for epoch in range(num_epochs):
-    print ("Epoch #",epoch)
+    print ("Epoch - ",epoch)
     model.train()
     optimizer.zero_grad()
     # Forward pass