search

fidelity / stoke

A lightweight wrapper for PyTorch that provides a simple declarative API for context switching between devices, distributed modes, mixed-precision, and PyTorch extensions.
https://fidelity.github.io/stoke/
Apache License 2.0
66 stars 3 forks source link

Bug fix for moving data to gpu #58

Closed ncilfone closed 2 years ago

ncilfone commented 2 years ago

forgot function return :eyes: