e5e7fb2f70cc8826fd624d1a3bc645fee3234d92,bootstrap/datasets/transforms.py,PadTensors,pad_tensors,#PadTensors#,47

Before Change

        if isinstance(batch, collections.Mapping):
            out = {}
            for key, value in batch.items():
                if key not in self.avoid_keys:
                    out[key] = self.pad_tensors(value)
                else:
                    out[key] = value

After Change

        if isinstance(batch, collections.Mapping):
            out = {}
            for key, value in batch.items():
                if (key in self.use_keys) or \
                   (len(self.use_keys) == 0 and key not in self.avoid_keys):
                    out[key] = self.pad_tensors(value)
                else:
                    out[key] = value

Instances