diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..a6d72e8
--- /dev/null
+++ b/environment.yml
@@ -0,0 +1,134 @@
+name: myenv
+channels:
+  - alex-marty
+  - pytorch
+  - conda-forge
+  - defaults
+dependencies:
+  - alembic=1.4.1=py_0
+  - appdirs=1.4.4=pyh9f0ad1d_0
+  - asn1crypto=1.4.0=pyh9f0ad1d_0
+  - blas=1.0=mkl
+  - brotlipy=0.7.0=py39hb82d6ee_1001
+  - ca-certificates=2021.4.13=haa95532_1
+  - certifi=2020.12.5=py39hcbf5309_1
+  - cffi=1.14.5=py39h0878f49_0
+  - chardet=4.0.0=py39hcbf5309_1
+  - click=8.0.1=py39hcbf5309_0
+  - cloudpickle=1.6.0=py_0
+  - colorama=0.4.4=pyh9f0ad1d_0
+  - configparser=5.0.2=pyhd8ed1ab_0
+  - cryptography=3.4.7=py39hd8d06c1_0
+  - cudatoolkit=10.2.89=h74a9793_1
+  - cycler=0.10.0=py39haa95532_0
+  - databricks-cli=0.9.1=py_0
+  - dataclasses=0.8=pyhc8e2a94_1
+  - docker-py=5.0.0=py39hcbf5309_0
+  - docker-pycreds=0.4.0=py_0
+  - docopt=0.6.2=py39haa95532_0
+  - entrypoints=0.3=pyhd8ed1ab_1003
+  - flask=2.0.1=pyhd8ed1ab_0
+  - freetype=2.10.4=hd328e21_0
+  - gitdb=4.0.7=pyhd8ed1ab_0
+  - gitpython=3.1.17=pyhd8ed1ab_0
+  - greenlet=1.1.0=py39h415ef7b_0
+  - icu=58.2=ha925a31_3
+  - idna=2.10=pyh9f0ad1d_0
+  - importlib-metadata=3.10.0=py39haa95532_0
+  - importlib_metadata=3.10.0=hd3eb1b0_0
+  - intel-openmp=2021.2.0=haa95532_616
+  - itsdangerous=2.0.1=pyhd8ed1ab_0
+  - jinja2=3.0.1=pyhd8ed1ab_0
+  - joblib=1.0.1=pyhd8ed1ab_0
+  - jpeg=9b=hb83a4c4_2
+  - jsonpickle=2.0.0=pyhd3eb1b0_0
+  - kiwisolver=1.3.1=py39hd77b12b_0
+  - libblas=3.9.0=1_h8933c1f_netlib
+  - libcblas=3.9.0=5_hd5c7e75_netlib
+  - liblapack=3.9.0=5_hd5c7e75_netlib
+  - libpng=1.6.37=h2a8f88b_0
+  - libprotobuf=3.17.1=h7755175_0
+  - libtiff=4.2.0=hd0e1b90_0
+  - libuv=1.40.0=he774522_0
+  - lz4-c=1.9.3=h2bbff1b_0
+  - m2w64-gcc-libgfortran=5.3.0=6
+  - m2w64-gcc-libs=5.3.0=7
+  - m2w64-gcc-libs-core=5.3.0=7
+  - m2w64-gmp=6.1.0=2
+  - m2w64-libwinpthread-git=5.0.0.4634.697f757=2
+  - mako=1.1.4=pyh44b312d_0
+  - markupsafe=2.0.1=py39hb82d6ee_0
+  - matplotlib=3.3.4=py39haa95532_0
+  - matplotlib-base=3.3.4=py39h49ac443_0
+  - mkl=2021.2.0=haa95532_296
+  - mkl-service=2.3.0=py39h2bbff1b_1
+  - mkl_fft=1.3.0=py39h277e83a_2
+  - mkl_random=1.2.1=py39hf11a4ad_2
+  - mlflow=1.17.0=py39h54a35c1_1
+  - msys2-conda-epoch=20160418=1
+  - munch=2.5.0=py_0
+  - ninja=1.10.2=h6d14046_1
+  - numpy=1.20.2=py39ha4e8547_0
+  - numpy-base=1.20.2=py39hc2deb75_0
+  - olefile=0.46=py_0
+  - openssl=1.1.1k=h8ffe710_0
+  - packaging=20.9=pyh44b312d_0
+  - pandas=1.2.4=py39hd77b12b_0
+  - pillow=8.2.0=py39h4fa10fc_0
+  - pip=21.1.1=py39haa95532_0
+  - prometheus_client=0.10.1=pyhd8ed1ab_0
+  - prometheus_flask_exporter=0.18.2=pyhd8ed1ab_0
+  - protobuf=3.17.1=py39h415ef7b_0
+  - py-cpuinfo=8.0.0=pyhd3eb1b0_0
+  - pycparser=2.20=pyh9f0ad1d_2
+  - pymongo=3.11.4=py39hd77b12b_0
+  - pyopenssl=20.0.1=pyhd8ed1ab_0
+  - pyparsing=2.4.7=pyhd3eb1b0_0
+  - pyqt=5.9.2=py39hd77b12b_6
+  - pysocks=1.7.1=py39hcbf5309_3
+  - python=3.9.5=h6244533_3
+  - python-dateutil=2.8.1=pyhd3eb1b0_0
+  - python-editor=1.0.4=py_0
+  - python-wget=3.2=py_0
+  - python_abi=3.9=1_cp39
+  - pytorch=1.8.1=py3.9_cuda10.2_cudnn7_0
+  - pytz=2021.1=pyhd3eb1b0_0
+  - pywin32=227=py39hb82d6ee_1
+  - pyyaml=5.4.1=py39hb82d6ee_0
+  - qt=5.9.7=vc14h73c81de_0
+  - querystring_parser=1.2.4=py_0
+  - requests=2.25.1=pyhd3deb0d_0
+  - sacred=0.7.4=py_0
+  - scikit-learn=0.24.2=py39he931e04_0
+  - scipy=1.6.3=py39hc0c34ad_0
+  - setuptools=52.0.0=py39haa95532_0
+  - sip=4.19.13=py39hd77b12b_0
+  - six=1.15.0=py39haa95532_0
+  - smmap=3.0.5=pyh44b312d_0
+  - sqlalchemy=1.4.15=py39hb82d6ee_0
+  - sqlite=3.35.4=h2bbff1b_0
+  - sqlparse=0.4.1=pyh9f0ad1d_0
+  - tabulate=0.8.9=pyhd8ed1ab_0
+  - threadpoolctl=2.1.0=pyh5ca1d4c_0
+  - tk=8.6.10=he774522_0
+  - torchaudio=0.8.1=py39
+  - torchvision=0.9.1=py39_cu102
+  - tornado=6.1=py39h2bbff1b_0
+  - typing_extensions=3.7.4.3=py_0
+  - tzdata=2020f=h52ac0ba_0
+  - urllib3=1.26.5=pyhd8ed1ab_0
+  - vc=14.2=h21ff451_1
+  - vs2015_runtime=14.27.29016=h5e58377_2
+  - waitress=2.0.0=pyhd8ed1ab_0
+  - websocket-client=0.57.0=py39hcbf5309_4
+  - werkzeug=2.0.1=pyhd8ed1ab_0
+  - wheel=0.36.2=pyhd3eb1b0_0
+  - win_inet_pton=1.1.0=py39hcbf5309_2
+  - wincertstore=0.2=py39h2bbff1b_0
+  - wrapt=1.12.1=py39h196d8e1_1
+  - xz=5.2.5=h62dcd97_0
+  - yaml=0.2.5=he774522_0
+  - zipp=3.4.1=pyhd3eb1b0_0
+  - zlib=1.2.11=h62dcd97_4
+  - zstd=1.4.5=h04227a9_0
+prefix: X:\Anaconda2020\envs\myenv