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