2022年3月11日
创建模型并且数据并行处理
摘要： 模型的实例，然后验证我们是否有多个 GPU。如果我们有多个 GPU，我们可以用 nn.DataParallel 来 包裹 我们的模型。然后我们使用 model.to(device) 把模型放到多 GPU 中。 model = Model(input_size, output_size) if tor 阅读全文
posted @ 2022-03-11 22:56 xjspyx 阅读(22) 评论(0) 推荐(0) 编辑