Renxi Wang

It is hard to locate the problem without details of the machine runing the code. There might be several possible reasons: different datasets, different cuda version, cpu bottleneck, databus bottleneck,...