[aisingapore#61] default model config

ktyap · Oct 25, 2022 · bbe2a00 · bbe2a00
1 parent 32dff5a
commit bbe2a00
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sgnlp/models/drnn_roberta/train.py b/sgnlp/models/drnn_roberta/train.py
@@ -178,7 +178,7 @@ def train_or_eval_model(model, loss_function, dataloader, epoch, optimizer=None,
 
     print(args)
 
-    model_path = pathlib.Path.cwd().parents[0].joinpath("temp/")
+    model_path = pathlib.Path.cwd().joinpath("temp/")
     print(model_path)
 
     global dataset