s440058/ium_440058
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

add pytorch exercise

Browse Source
This commit is contained in:
piotr6789 2021-04-25 23:49:57 +02:00
parent da8ea72364
commit 254da57d4b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch-example.py
View File

@ -48,7 +48,7 @@ criterion = torch.nn.BCELoss(reduction='mean')
optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate) optimizer = torch.optim.SGD(model.parameters(), lr = learning_rate)
for epoch in range(num_epochs): for epoch in range(num_epochs):
print ("Epoch #",epoch) print ("Epoch - ",epoch)
model.train() model.train()
optimizer.zero_grad() optimizer.zero_grad()
# Forward pass # Forward pass