是因为 我的model没有放在cuda上 原来是:
self.model = nn.DataParallel(network(**kwargs), device_ids=device_ids)
修改成 :
self.model = nn.DataParallel(network(**kwargs).cuda(), device_ids=device_ids)
是因为 我的model没有放在cuda上 原来是:
self.model = nn.DataParallel(network(**kwargs), device_ids=device_ids)
修改成 :
self.model = nn.DataParallel(network(**kwargs).cuda(), device_ids=device_ids)