The batch_size =4
criterion = nn.CrossEntropyLoss()
optimizer_ft = optim.SGD(model_ft.parameters(), lr=0.001, momentum=0.9)
Yes I’m normalize the data.
So, what do you think, i need increase the batch_size?