关于checkpoints的问题

162 阅读 0 评论 107 点赞

我是靠谱客的博主直率盼望，这篇文章主要介绍关于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 = [0，1，2]
net = nn.DataParallel(net, device_ids=device_ids)

其他地方无需改动，net在get_network已经放到cuda上了


if use_gpu:
net = net.cuda()
return net

net.cuda()

net = get_network(args)
#device_ids = [0，1，2]
#net = nn.DataParallel(net, device_ids=device_ids)

以上就是直率盼望最近收集整理的关于关于checkpoints的问题的全部内容，更多相关关于checkpoints内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供，作为学习参考使用，或来自网络收集整理，版权属于原作者所有。