diff --git a/train.py b/train.py
index ef049f0..e7ee137 100644
--- a/train.py
+++ b/train.py
@@ -9,7 +9,7 @@
 from data_utils import  calMetric_iou, LoadDatasetFromFolder
 from loss.BCL import BCL
 from loss.DiceLoss import DiceLoss
-from model.dsamnet import dsamnet
+from model.dsamnet import DSAMNet as dsamnet
 import numpy as np
 import random
 from train_options import parser