fix bugs in amazon-science#7

Author: Willy0919

lib/helpers/trainer_helper.py

@@ -8,7 +8,7 @@
 from lib.helpers.save_helper import get_checkpoint_state
 from lib.helpers.save_helper import save_checkpoint
 from lib.helpers.decorator_helper import decorator
-from lib.helpers.decorator_helper_level import decorator_level
+#from lib.helpers.decorator_helper_level import decorator_level
 
 
 
@@ -23,7 +23,7 @@ def __init__(self, cfg_trainer, model, optimizer, train_loader, test_loader,
         self.lr_scheduler = lr_scheduler
         self.bnm_scheduler = bnm_scheduler
         self.decorator = decorator
-        self.val_decorator = decorator_level
+#        self.val_decorator = decorator_level
         self.logger = logger
         self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
         self.model.to(self.device)