Thanks for sharing your code, the paper is really interesting. I hope to use it for future work and cite it!
I'm mostly posting this for visibility due to some issues that had occurred during setup and how I fixed them in my case:
On Ubuntu 20.04 with Anaconda 4.10.1,
during python setup.py build_ext --inplace we get the following ouput:
Traceback (most recent call last):
File "setup.py", line 7, in <module>
from torch.utils.cpp_extension import BuildExtension
File "/home/matt/anaconda3/envs/oflow/lib/python3.6/site-packages/torch/__init__.py", line 84, in <module>
from torch._C import *
ImportError: /home/matt/anaconda3/envs/oflow/lib/python3.6/site-packages/torch/lib/libmkldnn.so.0: undefined symbol: cblas_sgemm_alloc
For some reason pytorch doesn't install correctly so this can be solved with pip install torch==1.0.0.
And in setup.py include_dirs=[numpy_include_dir] needs to be added to each extension that requires numpy as it doesn't seem to find it otherwise.
Thanks for sharing your code, the paper is really interesting. I hope to use it for future work and cite it!
I'm mostly posting this for visibility due to some issues that had occurred during setup and how I fixed them in my case:
On Ubuntu 20.04 with Anaconda 4.10.1,
during
python setup.py build_ext --inplacewe get the following ouput:For some reason pytorch doesn't install correctly so this can be solved with
pip install torch==1.0.0.And in setup.py
include_dirs=[numpy_include_dir]needs to be added to each extension that requires numpy as it doesn't seem to find it otherwise.