Index 2700 is out of bounds for axis 0 with size 2700

6

Can you double check then that len(self.test_loader) is causing the problem?