State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reproducible accuracy and performance on enterprise-grade infrastructure.
13.61k
stars
3.24k
forks
source link
Fix device mismatch when using non-default GPUs. #1432
The
cudafunction will create the tensor on a default device that may differ from the device of tensor 'inverse_transform'.