[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**)
|