Skip to content

Instantly share code, notes, and snippets.

@ayulockin
Created May 8, 2020 14:57
Show Gist options
  • Save ayulockin/fc1ad1459d79a52658466a26e78715b9 to your computer and use it in GitHub Desktop.
Save ayulockin/fc1ad1459d79a52658466a26e78715b9 to your computer and use it in GitHub Desktop.
net = Net().to(device)
optimizer = optim.Adam(net.parameters())
wandb.init(project='pytorchw_b')
wandb.watch(net, log='all')
for epoch in range(10):
train(net, device, trainloader, optimizer, epoch)
test(net, device, testloader, classes)
print('Finished Training')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment