Puqing Deng

Thanks for your reply. Yes, I load my own features. I finally figure out that I need to remove the batch_size dimension in the input features.