7ee1d28458da68a76903a38dda5164e6abcaebf1,fairseq/trainer.py,Trainer,__init__,#Trainer#Any#Any#Any#,30

Before Change

        self.criterion = criterion.cuda()

        // initialize optimizer and LR scheduler
        self.optimizer = optim.build_optimizer(self.args, self.model.parameters())
        self.lr_scheduler = lr_scheduler.build_lr_scheduler(self.args, self.optimizer)

        // initialize meters
        self.meters = OrderedDict()
        self.meters["train_loss"] = AverageMeter()

After Change

        self.criterion = criterion.cuda()

        // initialize optimizer and LR scheduler
        self._build_optimizer()

        // initialize meters
        self.meters = OrderedDict()
        self.meters["train_loss"] = AverageMeter()

Instances