Merge pull request #8 from Willy0919/patch-3

fix bugs in #7

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)