Skip to content

Commit

Permalink
Updated naming of output_dir
Browse files Browse the repository at this point in the history
  • Loading branch information
MorenoLaQuatra committed Mar 2, 2023
1 parent 04b5d67 commit 4afa3e2
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion ic_finetuning.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,8 @@ def define_model(model_checkpoint, num_labels, label2id, id2label):
model_checkpoint = parse_cmd_line_params().model
gradient_accumulation_steps = parse_cmd_line_params().gradient_accumulation_steps
model_name = model_checkpoint.split("/")[-1]
output_dir = model_name + "-ic-finetuning"
dataset_name = dataset_name.split("italic-")[-1]
output_dir = model_name + "-ic-finetuning-" + dataset_name

## Load Dataset
dataset = load_dataset(
Expand Down

0 comments on commit 4afa3e2

Please sign in to comment.