Re:PyTorch 1.0 is out
by modeless on Friday December 07, 2018 @05:24PM
(#18632243)
Wow, this is cool. How complete is the API? Could you use it for research? Last I checked, the TensorFlow C++ API was missing all sorts of important stuff for building models and was basically only useful for loading models saved from Python.
[ Parent
| on HN ]
Yes! The API feels very much like using PyTorch from Python, and implementing models and working with tensors purely in C++ is very convenient. We're using it for our research platform for StarCraft: Brood War (https://torchcraft.github.io/TorchCraftAI).
[disclosure: I work at FAIR]
Re:PyTorch 1.0 is out
by goldsborough on Friday December 07, 2018 @08:35PM
(#18633283)
It's very complete. We have many kinds of builtin Modules (rnns, conv{1,2,3}d, batchnorm etc.), optimizers, a data loader, a serialization format for checkpointing and similar things you need for training. My hope is that any model you can train in Python you can now also train in C++.