Yalong Liu
Yalong Liu
For some reasons, I wrote my weights saving strategy which may be cause this error. However, I have fix the 'TypeError' today and you can download the new code now.
I can run the code with a free memory of about 10GB in GPU. Your free memory here is 5.13GB? May be there is no enough free memory in GPU...
I open the Anonfiles download link miniutes ago, and it does works! Moreover, I added GoogleDrive and BaiduYun download link in README.md today to prevent other accidents.