325d8412d641af7e52f97e3422e9ef516e4ef726,inferno/trainers/basic.py,Trainer,wrap_batch,#Trainer#Any#Any#Any#Any#,1028

Before Change

        // Cast to the right dtype
        batch = self.cast(batch)
        // Second, wrap as variable
        batch = type(batch)([Variable(_batch, requires_grad=requires_grad, volatile=volatile)
                             for _batch in batch])
        return batch

    def next_iteration(self):
        self._iteration_count += 1

After Change

        batch = self.cast(batch)
        // Second, wrap as variable
        variable_batch = []
        for batch_num, _batch in enumerate(batch):
            if thu.is_tensor(_batch):
                variable_batch.append(Variable(batch, requires_grad=requires_grad,
                                               volatile=volatile))
            elif pyu.is_listlike(_batch):
                variable_batch.append([Variable(__batch, requires_grad=requires_grad,
                                                volatile=volatile)
                                       for __batch in _batch])
            else:
                raise RuntimeError(f"Was Expecting batch at index {batch_num} to be either a "
                                   f"tensor or a list of tensors. Got {type(_batch)} instead.")
        batch = type(batch)(variable_batch)
        return batch

    def next_iteration(self):

Instances