diff --git a/ppgan/datasets/transforms/__init__.py b/ppgan/datasets/transforms/__init__.py
index 4d6aec073f4faf070c079ee144372f7dce53de0f..befc18568797cccb0670da028407797c02f247ef 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 71e568a43ff36c53fc5e30bc03f13dbd4fc49efd..e400e4bcd9023f678a16da172a04d93a47ba705f 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)