Suggestion: setup docker to be easy to develop.

[rrfaria]

If you put the project into a docker container it will be easier to run and develop and even to get more help

[lefnire]

Would love if someone did. I'm no longer working on project

[methenol]

Just uploaded a google colab notebook to the main directory. Copied the current code into the notebook and modified as little as possible to get it running, so it should give a good idea of how the code in the main repository works. Instructions for running are commented in the file but otherwise it's extremely barebone. This isn't a docker environment, but if you want a quick 'environment' setup to mess with the code, this is a pretty easy way and gets you a GPU to use. Should be able to load the github repo from inside of colab and select the notebook, recommend saving a copy in your google drive to save modifications.