From 4d36472b96d144887cbe95b083f0d2091fd5ff03 Mon Sep 17 00:00:00 2001 From: Julien Chaumond Date: Fri, 14 Feb 2020 03:25:29 +0000 Subject: [PATCH] [run_ner] Don't crash if fine-tuning local model that doesn't end with digit --- examples/run_ner.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/examples/run_ner.py b/examples/run_ner.py index a2937985ec..a1992d4df8 100644 --- a/examples/run_ner.py +++ b/examples/run_ner.py @@ -160,7 +160,10 @@ def train(args, train_dataset, model, tokenizer, labels, pad_token_label_id): # Check if continuing training from a checkpoint if os.path.exists(args.model_name_or_path): # set global_step to gobal_step of last saved checkpoint from model path - global_step = int(args.model_name_or_path.split("-")[-1].split("/")[0]) + try: + global_step = int(args.model_name_or_path.split("-")[-1].split("/")[0]) + except ValueError: + global_step = 0 epochs_trained = global_step // (len(train_dataloader) // args.gradient_accumulation_steps) steps_trained_in_current_epoch = global_step % (len(train_dataloader) // args.gradient_accumulation_steps)