amjltc295 / pytorch-golden-template

PyTorch Golden Template (under development)
13 stars 4 forks source link

Global device #35

Closed amjltc295 closed 5 years ago

amjltc295 commented 5 years ago

Loss/metrics may need to get the device (e.g. Perceptual loss), so we may want to make it global