models

[aprillysea]

Traceback (most recent call last):
File "visualize/generate.py", line 26, in
from models.ce import make_model
ModuleNotFoundError: No module named 'models'

[niuniulla]

This is because when you call 'generate.py' in the directory 'visualize', the path 'models/' is not exposed in the python path. So either add models in your python path either in the code directly using 'sys' or by modifying your environment variable manually (eg. 'PYTHONPATH="$PWD" python visualize/generate.py ....' if you are in the project parent directory).