This repository was archived by the owner on Dec 2, 2024. It is now read-only.
This repository was archived by the owner on Dec 2, 2024. It is now read-only.
return x.cuda(async=True) if torch.cuda.is_available() else x #35
async is now a keyword. It should be changed to 'non_blocking'