Postprocess.py

class Postprocess:
  
  def per_batch_transform(self, batch):
    # logic to postprocess a prediction batch
  
  def uncollate(self, batch):
    # logic to convert a prediction batch into a sequence of predictions
  
  def per_sample_transform(self, samples)
    # logic to postprocess a sequence of predictions