diff --git a/sgnlp/models/drnn_roberta/train.py b/sgnlp/models/drnn_roberta/train.py index 19edc0e..2577ab2 100644 --- 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