2154d789ef0d79debe2cc522751ae7072e8b22dc,torchsample/modules/super_module.py,SuperModule,predict_on_batch,#SuperModule#Any#Any#,492

Before Change

        if cuda_device > 0:
            x = x.cuda(cuda_device)
        preds = self(x)
        self.train()
        return preds

    def evaluate(self, 
                 inputs, 

After Change

    def predict_on_batch(self, 
                         inputs, 
                         cuda_device=-1):
        if not isinstance(inputs, (list,tuple)):
            inputs = [inputs]
        if cuda_device > -1:
            inputs = [ins.cuda(cuda_device) for ins in inputs]
        preds = self(*inputs)
        return preds

Instances