transfer-learning-conv-ai
transfer-learning-conv-ai copied to clipboard
Error while saving models
Hi,
Thank you for sharing us the code. I am using this repo to fine-tune GPT2 model on my conversation dataset. I am using pytorch_transformers to import AdamW, OpenAIGPTDoubleHeadsModel, OpenAIGPTTokenizer, GPT2DoubleHeadsModel, GPT2Tokenizer, WEIGHTS_NAME. I get this error after it runs all epochs.
Traceback (most recent call last):
File "./train_no_sc.py", line 283, in
Can you please suggest me?
The issue is resolved when pytorch-ignite==1.2.1 is used.