From f2fee9179c8688de7611561596e0c4cc7e38f9c6 Mon Sep 17 00:00:00 2001 From: alexcbb <alexchapin@hotmail.fr> Date: Wed, 12 Jul 2023 14:38:44 +0200 Subject: [PATCH] Fix import issues --- train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/train.py b/train.py index 40a1412..c4a7140 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 -- GitLab