sheepsky

Results 4 comments of sheepsky

请问用的哪个版本的torc和cuda

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...