关于多卡训练的问题

[tlzhaoboling]

您好，关于多卡的训练，我修改了两处地方，包括如下： os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3"
def train(train_loader, model, optimizer, epoch): model = torch.nn.DataParallel(model, device_ids=[0, 1, 2]).cuda() #添加 model.train() [训练后的模型经过测试发现结果全是灰色图，如下

想问下您是否知道出了什么问题呢，万分感谢

[thograce]

您好，关于多卡的训练，我修改了两处地方，包括如下： os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3" def train(train_loader, model, optimizer, epoch): model = torch.nn.DataParallel(model, device_ids=[0, 1, 2]).cuda() #添加 model.train() [训练后的模型经过测试发现结果全是灰色图，如下 想问下您是否知道出了什么问题呢，万分感谢

你要在主函数112行那里改，不要在train()函数里面加这些代码。

[tlzhaoboling]

您好，关于多卡的训练，我修改了两处地方，包括如下： os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3" def train(train_loader, model, optimizer, epoch): model = torch.nn.DataParallel(model, device_ids=[0, 1, 2]).cuda() #添加 model.train() [训练后的模型经过测试发现结果全是灰色图，如下 想问下您是否知道出了什么问题呢，万分感谢

你要在主函数112行那里改，不要在train()函数里面加这些代码。

您好，关于多卡的训练，我修改了两处地方，包括如下： os.environ["CUDA_VISIBLE_DEVICES"] = "0,1,2,3" def train(train_loader, model, optimizer, epoch): model = torch.nn.DataParallel(model, device_ids=[0, 1, 2]).cuda() #添加 model.train() [训练后的模型经过测试发现结果全是灰色图，如下 想问下您是否知道出了什么问题呢，万分感谢

你要在主函数112行那里改，不要在train()函数里面加这些代码。

您好，根据您的提示，我在112行那里改，仍出现相同的问题，以及我根据参考您提供的训练代码修改，仍是如此，想知道是什么问题导致的呢，参数的问题吗还是哪里？谢谢您

[thograce]

您好，根据您的提示，我在112行那里改，仍出现相同的问题，以及我根据参考您提供的训练代码修改，仍是如此，想知道是什么问题导致的呢，参数的问题吗还是哪里？谢谢您

你把微信发送到我邮箱吧，我给你看看代码，thograce@163.com。

[thograce]

您好，根据您的提示，我在112行那里改，仍出现相同的问题，以及我根据参考您提供的训练代码修改，仍是如此，想知道是什么问题导致的呢，参数的问题吗还是哪里？谢谢您

大概率是你模型输出的结果没有激活，但我的代码里是有的，你是不是删了什么？建议你把我训练好的模型下载下来后直接运行etest.py，看输出结果是否正常。

[tlzhaoboling]

其它的地方我是没有改的，我拿单卡训练，测试结果是没有问题的，但一多卡训练，测试出来的结果就是这样