1 files changed, 2 insertions, 2 deletions
diff --git a/train.py b/train.py
index ee8cff1..dd79bab 100644
--- a/train.py
+++ b/train.py
@@ -1,7 +1,7 @@
 import time
 from options.train_options import TrainOptions
-from data.data_loader import CreateDataLoader
-from models.models import create_model
+from data import CreateDataLoader
+from models import create_model
 from util.visualizer import Visualizer
 
 if __name__ == '__main__':