GengDavid/pytorch-cpn

target.cuda(async=True),在async=True处出现SyntaxError: invalid syntax

my-hello-world opened this issue · 1 comments

my-hello-world commented

相关资料不是很直观，想问下我直接把

        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())
Crophone commented

yes.you can use the .cuda()
here is the reference