【警告】[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool after fork. (function pthreadpool

[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool after fork. (function pthreadpool)
[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool after fork. (function pthreadpool)
[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool after fork. (function pthreadpool)
[W pthreadpool-cpp.cc:90] Warning: Leaking Caffe2 thread-pool after fork. (function pthreadpool)

找到:

torch.utils.data.DataLoader(
            val_dataset, batch_size=args.batch_size, shuffle=True, num_workers=args.num_workers, 
            pin_memory=**True**)

改为

torch.utils.data.DataLoader(
       	val_dataset, batch_size=args.batch_size, shuffle=True, num_workers=args.num_workers, 
       	pin_memory=**False**)

你可能感兴趣的:(pytorch,深度学习)