摘要:
数据并行(选读) Authors: Sung Kim and Jenny Kang 在这个教程里,我们将学习如何使用 DataParallel 来使用多GPU。 PyTorch非常容易就可以使用多GPU,用如下方式把一个模型放到GPU上: device = torch.device("cuda:0" 阅读全文
摘要:
文章目录 训练一个分类器关于数据?训练一个图像分类器在GPU上训练多GPU训练下一步? 训练一个分类器 上一讲中已经看到如何去定义一个神经网络,计算损失值和更新网络的权重。 你现在可能在想下一步。 关于数据? 一般情况下处理图像、文本、音频和视频数据时,可以使用标准的Python包来加载数据到一个n 阅读全文