VL-PLM icon indicating copy to clipboard operation
VL-PLM copied to clipboard

Published 20 hours ago • verified publisher icon xiaofeng94

您好，进行训练必须在多gpu的环境下嘛

Open minC0819 opened this issue 11 months ago • 3 comments

minC0819 avatar Feb 28 '24 11:02 minC0819

Hey, detectron2也支持单GPU训练，但我们只在多GPU上进行过训练。改变GPU数量需要相应改变学习率，batch size和梯度累加策略等来保证复现论文中的结果。

xiaofeng94 avatar Feb 28 '24 19:02 xiaofeng94

嘿，Detectron2也支持单GPU训练，但我们训练只在多GPU上进行过。改变GPU数量需要相应的学习率，batch size和改变梯度累加策略等来保证复现论文中的结果。

minC0819 avatar Feb 29 '24 02:02 minC0819

您好，出现了RuntimeError: Default process group has not been initialized, please make sure to call init_process_group.这种问题是因为没使用多GPU的问题嘛

minC0819 avatar Feb 29 '24 02:02 minC0819