model = VideoClassifier() # Assuming you have your data loader and device (GPU/CPU) device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") model.to(device)