28ec5687220f562b3bd1b220ed80eded79c54824,steps/pytorch/models.py,Model,_transform,#Model#Any#Any#,87

Before Change

            else:
                X = Variable(X)
            output = self.model(X)
            outputs.append(output.data.cpu().numpy())

            if batch_id == steps:
                break

After Change

            outputs_batch = self.model(X)
            if len(self.output_names) == 1:
                outputs.setdefault(self.output_names[0], []).append(outputs_batch.data.cpu().numpy())
            else:
                for name, output in zip(self.output_names, outputs_batch):
                    output_ = output.data.cpu().numpy()
                    outputs.setdefault(name, []).append(output_)

Instances