Method step

TinyViT/tinyvit_utils.py:35–37

(self, epoch)

        self.optimizer = optimizer

    def step(self, epoch):
        self.lr_scheduler.step(epoch)
        self.update_lr()

    def step_update(self, it):
        self.lr_scheduler.step_update(it)