Method set_step

src/utils/visualization.py:28–30

(self, step, mode='train')

        ]

    def set_step(self, step, mode='train'):
        self.mode = mode
        self.step = step

    def __getattr__(self, name):
        """