File "main.py", line 161
target = target.cuda(async=True)
^
SyntaxError: invalid syntax
解决方法:
将async改为non_blocking即可
File "main.py", line 161
target = target.cuda(async=True)
^
SyntaxError: invalid syntax
解决方法:
将async改为non_blocking即可