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