updated with pull requests, fixed logging

gorkemgoknar · Dec 21, 2020 · 898c902 · 898c902
1 parent af3bab1
commit 898c902
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/train.py b/train.py
@@ -189,7 +189,6 @@ def train():
     tokenizer_class = GPT2Tokenizer if "gpt2" in args.model_checkpoint else OpenAIGPTTokenizer # cant use Autotokenizer because checkpoint could be a Path
     tokenizer = tokenizer_class.from_pretrained(args.model_checkpoint)
 
-
     model_class = GPT2DoubleHeadsModel if "gpt2" in args.model_checkpoint else OpenAIGPTDoubleHeadsModel
     model = model_class.from_pretrained(args.model_checkpoint)
     model.to(args.device)