updated docs to show how to use a LR scheduler. cleaned up some code.

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 docs to show how to use a LR scheduler. cleaned up some code. #30

Closed ncilfone closed 2 years ago