diff --git a/train.py b/train.py
index 64a9ca9..5cfaf61 100644
--- a/train.py
+++ b/train.py
@@ -9,7 +9,7 @@
 import numpy as np
 import torch
 from torch.optim import Adam
-
+import os, sys
 BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
 sys.path.append(BASE_DIR + "/multiwoz/model/")