If anyone wants to play around with this in a hosted gpu supported jupyter notebook, check out: https://colab.research.google.com/github/AliaksandrSiarohin/fir

If anyone wants to play around with this in a hosted gpu supported jupyter notebook, check out: https://colab.research.google.com/github/AliaksandrSiarohin/first-order-model/blob/master/demo.ipynb https://colab.research.google.com/github/AliaksandrSiarohin/... I think this is just a wrapper of that script