模型训练到一个新batch时会被killed【原因以及解决方法】

模型训练到一个新batch时会被killed【原因以及解决方法】:
1.减小batchsize
2.减小num_workers(在Dataloader函数中)

猜你喜欢

转载自blog.csdn.net/qq_43663979/article/details/129399948
今日推荐