AIX360 icon indicating copy to clipboard operation
AIX360 copied to clipboard

Published 20 hours ago • verified publisher icon Trusted-AI

Optimize torch.load for CPU and GPU usage

Open dheckl opened this issue 5 years ago • 1 comments

Auto-select the first CUDA-device if CUDA is available.

dheckl avatar Feb 03 '20 15:02 dheckl

@pronics2004 Could you please review this change?

vijay-arya avatar Apr 05 '20 11:04 vijay-arya