train.py

@@ -38,7 +38,7 @@ def main():
     with open('class_indices.json', 'w') as json_file:
         json_file.write(json_str)
 
-    batch_size = 128
+    batch_size = 64
     nw = min([os.cpu_count(), batch_size if batch_size > 1 else 0, 8])  # number of workers
     print('Using {} dataloader workers every process'.format(nw))
 
@@ -77,7 +77,7 @@ def main():
     loss_function = nn.CrossEntropyLoss()
     optimizer = optim.Adam(net.parameters(), lr=0.0003)
 
-    epochs = 25
+    epochs = 40
     best_acc = 0.0
     # save_path = './googleNet.pth'
     train_steps = len(train_loader)