diff --git a/train.py b/train.py
index 40a14127d066a4e2548e5969ad57acdd83b64191..c4a71403784d317d5818aa6748c1186bc1ef32d3 100755
--- a/train.py
+++ b/train.py
@@ -17,7 +17,7 @@ import yaml
 
 from osrt import data
 from osrt.model import OSRT
-from osrt.trainer import SRTTrainer, OSRTSamTrainer
+from osrt.trainer import SRTTrainer
 from osrt.layers import Transformer
 from osrt.checkpoint import Checkpoint
 from osrt.utils.common import init_ddp