1 files changed, 4 insertions, 10 deletions

diff --git a/trainer_old/base.py b/trainer_old/base.py
index 1f85e71..5903d96 100644
--- a/trainer_old/base.py
+++ b/trainer_old/base.py
@@ -174,19 +174,13 @@ class TrainingStrategy():
 
     @contextmanager
     def on_train(self, epoch: int):
-        try:
-            self.tokenizer.train()
-            yield
-        finally:
-            pass
+        self.tokenizer.train()
+        yield
 
     @contextmanager
     def on_eval(self):
-        try:
-            self.tokenizer.eval()
-            yield
-        finally:
-            pass
+        self.tokenizer.eval()
+        yield
 
     def on_before_optimize(self, epoch: int):
         ...