Met 'Value Error' when train my own data

[STAR-811]

Hi! Thank you for offering the code.
When I try to train the HDN network on my own microscopy data (data type: int16 data range: -179 to 20351), at first everything goes well. But when it comes to 32 epochs, it reports 'Value Error':

Using the same training parameters, the Convallaria dataset didn't encounter this problem. I would appreciate it if you could help me!