Xinheng He (何欣恒)

I also found that it has issues with input data, batch size and its hyperparameters. Struggled for 1 week for running without nan but fails. Maybe such network only works...

 I made it by adding a module in trainind_step like this, however, when I run the code on multiple GPUs, it stops when return None (no matter clean or...