I totally understand that it is hard to debug without looking at the code at least. I was thinking maybe this was a known issue with a known solution. I will post soon the code. It is a somewhat complicated pipeline and you won’t be able to run the training because the data comes from digital pathology images. Hopefully the code will illuminate on the issue. Thanks in advance.