Skip to content

Commit

Permalink
corrected paths
Browse files Browse the repository at this point in the history
  • Loading branch information
davda54 authored Dec 5, 2021
1 parent f59f215 commit cb2e3cc
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions train.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,7 @@
pl.seed_everything(args.seed)

timestamp = f"{datetime.datetime.today():%m-%d-%y_%H-%M-%S}"
directory = f"/lnet/work/people/samuel/outputs/{args.dataset.language}_{args.model.pretrained_lm.split('/')[-1]}_{timestamp}"
directory = f"outputs/{args.dataset.language}_{args.model.pretrained_lm.split('/')[-1]}_{timestamp}"
predictions_directory = f"{directory}/test_predictions"
os.mkdir(directory)
os.mkdir(predictions_directory)
Expand Down Expand Up @@ -58,7 +58,7 @@
# INFERENCE ON TEST

def inference(args, model, input_data, output_dir):
args.dataset.path = f"/home/samuel/personal_work_ms/w-nut-normalization/data/multilexnorm/test-eval/test/{input_data}"
args.dataset.path = f"data/multilexnorm/test-eval/test/{input_data}"
if not os.path.isdir(output_dir):
os.mkdir(output_dir)

Expand Down

0 comments on commit cb2e3cc

Please sign in to comment.