clara/pytorch-ai
1
0
Fork 0
Code Issues 1 Pull requests Projects Releases Packages Wiki Activity Actions

Added chapters, trained net

Browse source
This commit is contained in:
Clemens Dautermann 2019-12-27 20:08:35 +01:00
parent 9350b1e4ba
commit a3f984996c
30 changed files with 1134 additions and 934 deletions
Download patch file Download diff file Expand all files Collapse all files

4
TicTacToe_AI/Net/pytorch_ai.py
View file

@ -79,7 +79,7 @@ class Net(torch.nn.Module):
return F.log_softmax(x, dim=1)
net = Net()
net = torch.load('./nets/net_3.pt')
optimizer = optim.Adam(net.parameters(), lr=0.001)
@ -95,5 +95,5 @@ for epoch in range(100):
optimizer.step()
print(loss)
torch.save(net, './nets/net_' + str(epoch) + '.pt')
torch.save(net, './nets/net_' + str(epoch + 3) + '.pt')
testnet(net, testset)