requirements.txt

pillow<7  #  remove this when moving to torchvision>0.5 (see https://github.com/pytorch/vision/issues/1712#issuecomment-575036523)
torch==1.3.1
numpy>=1.16
torchvision==0.4.2
scipy>=1.3.0
gitpython==3.1.0
torchnet==0.0.4
tensorboardx
pydot==1.4.1
tabulate==0.8.3
pandas>=0.22.0
jupyter>=1.0.0
matplotlib~=3.0  # 3.0 is the last release to support Py3.5
qgrid==1.1.1
graphviz==0.10.1
ipywidgets==7.4.2
bqplot==0.11.5
pyyaml
pytest~=4.6.1
xlsxwriter>=1.2.2
pretrainedmodels==0.7.4
scikit-learn==0.21.2
gym==0.12.5
tqdm==4.33.0