Add "allow_pickle=True"

chenyuntc / pytorch-book

PyTorch tutorials and fun projects including neural talk, neural style, poem writing, anime generation (《深度学习框架PyTorch：入门与实战》)

MIT License

12.11k stars 3.74k forks source link

Closed lucifer1004 closed 4 years ago

lucifer1004 commented 5 years ago

This makes it compatible with the latest numpy, otherwise, there would be an Object arrays cannot be loaded when allow_pickle=False error.

chenyuntc commented 4 years ago

Thanks @lucifer1004