From 355e55657df03aa843680b3b4ad8860166b2dc8c Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Thu, 22 Oct 2020 08:28:31 +0000 Subject: [PATCH] clean code --- ppgan/datasets/transforms/__init__.py | 1 - ppgan/engine/trainer.py | 10 ---------- 2 files changed, 11 deletions(-) diff --git a/ppgan/datasets/transforms/__init__.py b/ppgan/datasets/transforms/__init__.py index 4d6aec0..befc185 100644 --- a/ppgan/datasets/transforms/__init__.py +++ b/ppgan/datasets/transforms/__init__.py @@ -1,2 +1 @@ -# from .transforms import RandomCrop, Resize, RandomHorizontalFlip, PairedRandomCrop, PairedRandomHorizontalFlip, Normalize, Permute from .transforms import PairedRandomCrop, PairedRandomHorizontalFlip diff --git a/ppgan/engine/trainer.py b/ppgan/engine/trainer.py index 71e568a..e400e4b 100644 --- a/ppgan/engine/trainer.py +++ b/ppgan/engine/trainer.py @@ -270,22 +270,12 @@ class Trainer: if state_dicts.get('epoch', None) is not None: self.start_epoch = state_dicts['epoch'] + 1 - # for name in self.model.model_names: - # if isinstance(name, str): - # net = getattr(self.model, 'net' + name) - # net.set_dict(state_dicts['net' + name]) - for net_name, net in self.model.nets.items(): net.set_dict(state_dicts[net_name]) for opt_name, opt in self.model.optimizers.items(): opt.set_dict(state_dicts[opt_name]) - # for name in self.model.optimizer_names: - # if isinstance(name, str): - # opt = getattr(self.model, name) - # opt.set_dict(state_dicts[name]) - def load(self, weight_path): state_dicts = load(weight_path) -- GitLab