Update train.py

filippovella · Dec 5, 2024 · 1d10b8a · 1d10b8a
1 parent 73c7a9a
commit 1d10b8a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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/")