pytorch-cpn
pytorch-cpn copied to clipboard
target.cuda(async=True),在async=True处出现SyntaxError: invalid syntax
相关资料不是很直观,想问下我直接把
refine_target_var = torch.autograd.Variable(target7.cuda(async=True))
valid_var = torch.autograd.Variable(valid.cuda(async=True))
改成如下可以吗?
refine_target_var = torch.autograd.Variable(target7.cuda())
valid_var = torch.autograd.Variable(valid.cuda())
yes.you can use the .cuda() here is the reference