Skip to content

Commit

Permalink
Update README.md (#254)
Browse files Browse the repository at this point in the history
Add skorch to README
  • Loading branch information
ethanwharris authored and MattPainter01 committed Jul 25, 2018
1 parent 7c88f1a commit e196bbf
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -126,7 +126,8 @@ Our documentation containing the API reference, examples and some notes can be f

## Other Libraries

Torchbearer isn't the only library for training PyTorch models. Here are a few others that might better suit your needs:
Torchbearer isn't the only library for training PyTorch models. Here are a few others that might better suit your needs (this is by no means a complete list, see the [awesome pytorch list](https://github.com/bharathgs/Awesome-pytorch-list) for more):
- [skorch](https://github.com/dnouri/skorch), model wrapper that enables use with scikit-learn - crossval etc. can be very useful
- [PyToune](https://github.com/GRAAL-Research/pytoune), simple Keras style API
- [ignite](https://github.com/pytorch/ignite), advanced model training from the makers of PyTorch, can need a lot of code for advanced functions (e.g. Tensorboard)
- [TorchNetTwo (TNT)](https://github.com/pytorch/tnt), can be complex to use but well established, somewhat replaced by ignite

0 comments on commit e196bbf

Please sign in to comment.