关于checkpoints的问题
关于单与多GPU的checkpoints的问题经过验证,多GPU训练并保存的checkpoints与单GPU checkpoints有区别验证用四个gpu【0,1,2,3】训练的模型可以用【0,1,2】三个gpu来test,也可以用【0,1,2,3,5】五个gpu,甚至可以用【0】单个gpu来验证,需要将代码设置为以下:net = get_network(args)device_ids =...