sheepsky
Results
4
comments of
sheepsky
Thanks for your replying
请问用的哪个版本的torc和cuda
how to slolve this?
I have solved this, we should replace send_to_device function to the following(not same with the solution in Common Issue 3): ``` def send_to_device(tensor, device, non_blocking=False, skip_keys=None): """ Recursively sends the...