Added chapters, trained net

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

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

Binary file not shown.

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)