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

Updated Dep Versions #51

Closed ncilfone closed 2 years ago

ncilfone commented 2 years ago

What does this PR do?

Bumped versions of deepseep and fairscale. adjusted configs to deal with new configuration options from both libraries

Checklist