Method to

mGPT/data/transforms/base.py:66–70

(self, *args, **kwargs)

        return iter(data)

    def to(self, *args, **kwargs):
        for key in self.datakeys:
            if self[key] is not None:
                self[key] = self[key].to(*args, **kwargs)
        return self

    @property
    def device(self):