def _init_optims(self, lr=0.0003, **kwargs):
    super()._init_optims(**kwargs)
    self.register_optimizer(torch.optim.Adam(self.model.parameters(), lr=lr, amsgrad=True), DABaselineLoss(self))
def _init_optims(self, lr=0.0003, **kwargs):
    super()._init_optims(**kwargs)
    self.register_optimizer(torch.optim.Adam(self.model.parameters(), lr=lr, amsgrad=True), DABaselineLoss(self))