yet-another-retnet
yet-another-retnet copied to clipboard
ModelCheckpoint() argument after ** must be a mapping, not ModelCheckpoint
I tried to do inference but got the above error. I'm using python3.10 on Ubuntu 22.04. i9-13900K + 4090 I fixed this by changing return cls(**checkpoint_dict) to return cls(**vars(checkpoint_dict))
File "/home/dwood/LLM/yar/./venv/lib/python3.10/site-packages/scripts/train_project_gutenberg.py", line 95, in load
return cls(**checkpoint_dict)
TypeError: __main__.ModelCheckpoint() argument after ** must be a mapping, not ModelCheckpoint