liyuzhen

在单卡GPU上训练出现的问题，无法解决

3

作者您好，我用的Imagenet-mini数据集，在单卡GPU上训练。看到您之前回答的问题：在单卡GPU上训练，需要将lr减小到原来的8倍，这样设置：--nproc_per_node=1、--lr 2.5e-4。但是我尝试了这样设置的前提下，batchsize=128 batchsize=64 batchsize=32 batchsize=16 代码没有任何修改的情况下，都会出现不同程度的问题。