having error in
loss = criterion(output, target) which does not runs.
but loss previously made is not an integer
here is my criterion
criterion = nn.CrossEntropyLoss()
optimizer = optim.Adam(model_scratch.parameters())