Pytorch dataloader 设置 num_worker

2020-01-31  本文已影响0人  Williamongh

https://www.codeleading.com/article/7316314042/
https://www.jianshu.com/p/4a1a92f0efd9

if __name__ == '__main__':应该加在最前端, 不然均会有问题 (在文件最开端写一句print('Hello World!'), 会多次输出)

上一篇 下一篇

猜你喜欢

热点阅读