From 55ced5c633eabfd835caf17152cf75a82752810c Mon Sep 17 00:00:00 2001 From: jscsmk Date: Fri, 11 Oct 2019 18:25:09 +0900 Subject: [PATCH] Fix typo modle to model --- src/lib/trains/base_trainer.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/lib/trains/base_trainer.py b/src/lib/trains/base_trainer.py index 3ff6efb..1f29865 100755 --- a/src/lib/trains/base_trainer.py +++ b/src/lib/trains/base_trainer.py @@ -9,9 +9,9 @@ from models.data_parallel import DataParallel from utils.utils import AverageMeter -class ModleWithLoss(torch.nn.Module): +class ModelWithLoss(torch.nn.Module): def __init__(self, model, loss): - super(ModleWithLoss, self).__init__() + super(ModelWithLoss, self).__init__() self.model = model self.loss = loss @@ -26,7 +26,7 @@ class BaseTrainer(object): self.opt = opt self.optimizer = optimizer self.loss_stats, self.loss = self._get_losses(opt) - self.model_with_loss = ModleWithLoss(model, self.loss) + self.model_with_loss = ModelWithLoss(model, self.loss) def set_device(self, gpus, chunk_sizes, device): if len(gpus) > 1: