This error pops up while training the skin cancer model. Changed the batch size to 1 but it didn't solve the problem.