Chin_GM:

alidation


First of all I will advise you remove the dropout as it is usually used for large networks and try using an Adam Optimizer optimizer = torch.optim.Adam(net.parameters(), lr=0.001)