Bug Issue: drop_last=True for test dataset lead to inaccurate test results

[huangst21]

Hi, thanks to the authors for their valuable work, the results of this work are amazing. However, I think potential bugs may have lead to the incorrect performance of this work. As https://github.com/vewoxic/fits has shown, using drop=True during the testing phase can lead to incorrect results. In this work, the batch size was set to 1024, and this large batch size may have amplified the impact of the bug. Hope the authors will fix the bug and correct the experimental results.