code: def predict_batch(self, batch):
      return self.model(
          batch[0].to(self.device).view(batch[0].size(0), -1)
      )
  
file_path: fastai/nbs/examples/migrating_catalyst.py
imports: from fastai.data.load import DataLoader
  import os, torch
  from torch.nn import functional
  from catalyst import dl
  from catalyst.data.cv import ToTensor
  from catalyst.contrib.datasets import MNIST
  from catalyst.utils import metrics
label_desc: Use image node_id 1 for calling the CustomRunner obj's underlying member method code with example usage: obj.predict_batch(batch) and returns: self