出现这个错误的情况是,在服务器上的docker中运行训练代码时,batch size设置得过大,shared memory不够(因为docker限制了shm).

解决方法是,找到代码中的Dataloader将num_workers设置为0.

loader = DataLoader(dset, batch_size=args.batch, num_workers=0)
Logo

权威|前沿|技术|干货|国内首个API全生命周期开发者社区

更多推荐