def configure_optimizers(self):
    return torch.optim.Adam(self.parameters(), lr=self.cfg.LEARNER.LEARNING_RATE)
def configure_optimizers(self):
    return torch.optim.Adam(self.parameters(), lr=self.cfg.LEARNER.LEARNING_RATE)