diff --git a/environment.yml b/environment.yml
new file mode 100644
index 0000000..00b25ae
--- /dev/null
+++ b/environment.yml
@@ -0,0 +1,211 @@
+name: iumenv
+channels:
+  - alex-marty
+  - defaults
+  - conda-forge
+  - pytorch
+dependencies:
+  - _libgcc_mutex=0.1=main
+  - _openmp_mutex=4.5=1_gnu
+  - _pytorch_select=0.1=cpu_0
+  - _tflow_select=2.3.0=mkl
+  - abseil-cpp=20200923.3=h2531618_0
+  - absl-py=0.12.0=py39h06a4308_0
+  - aiohttp=3.7.4=py39h27cfd23_1
+  - alembic=1.3.1=py_0
+  - alsa-lib=1.2.3=h516909a_0
+  - appdirs=1.4.4=py_0
+  - asn1crypto=1.4.0=py_0
+  - astor=0.8.1=py39h06a4308_0
+  - astunparse=1.6.3=py_0
+  - async-timeout=3.0.1=py39h06a4308_0
+  - attrs=21.2.0=pyhd3eb1b0_0
+  - blas=1.0=mkl
+  - blinker=1.4=py39h06a4308_0
+  - c-ares=1.17.1=h27cfd23_0
+  - ca-certificates=2021.5.25=h06a4308_1
+  - cached-property=1.5.2=py_0
+  - cachetools=4.2.2=pyhd3eb1b0_0
+  - certifi=2021.5.30=py39h06a4308_0
+  - cffi=1.14.5=py39h261ae71_0
+  - chardet=3.0.4=py39h06a4308_1003
+  - click=8.0.1=pyhd3eb1b0_0
+  - cloudpickle=1.6.0=py_0
+  - configparser=5.0.2=pyhd3eb1b0_0
+  - coverage=5.5=py39h27cfd23_2
+  - cryptography=3.4.7=py39hd23ed53_0
+  - cudatoolkit=10.2.89=hfd86e86_1
+  - cycler=0.10.0=py39h06a4308_0
+  - cython=0.29.23=py39h2531618_0
+  - databricks-cli=0.9.1=py_0
+  - dbus=1.13.18=hb2f20db_0
+  - docker-py=4.4.1=py39h06a4308_5
+  - docker-pycreds=0.4.0=py_0
+  - docopt=0.6.2=py39h06a4308_0
+  - entrypoints=0.3=py39h06a4308_0
+  - expat=2.4.1=h2531618_2
+  - flask=1.1.2=pyhd3eb1b0_0
+  - flatbuffers=2.0.0=h2531618_0
+  - fontconfig=2.13.1=h6c09931_0
+  - freetype=2.10.4=h5ab3b9f_0
+  - gast=0.4.0=py_0
+  - gettext=0.21.0=hf68c758_0
+  - giflib=5.2.1=h7b6447c_0
+  - gitdb=4.0.7=pyhd3eb1b0_0
+  - gitpython=3.1.17=pyhd3eb1b0_1
+  - glib=2.68.2=h36276a3_0
+  - google-auth=1.30.1=pyhd3eb1b0_0
+  - google-auth-oauthlib=0.4.1=py_2
+  - google-pasta=0.2.0=py_0
+  - greenlet=1.1.0=py39h2531618_0
+  - grpcio=1.36.1=py39h2157cd5_1
+  - gst-plugins-base=1.18.4=hf529b03_2
+  - gstreamer=1.18.4=h76c114f_2
+  - gunicorn=20.1.0=py39h06a4308_0
+  - h5py=3.2.1=py39h6c542dc_0
+  - hdf5=1.10.6=hb1b8bf9_0
+  - icu=68.1=h2531618_0
+  - idna=2.10=pyhd3eb1b0_0
+  - importlib-metadata=3.10.0=py39h06a4308_0
+  - importlib_metadata=3.10.0=hd3eb1b0_0
+  - intel-openmp=2021.2.0=h06a4308_610
+  - itsdangerous=2.0.1=pyhd3eb1b0_0
+  - jinja2=3.0.0=pyhd3eb1b0_0
+  - jpeg=9d=h36c2ea0_0
+  - jsonpickle=2.0.0=pyhd3eb1b0_0
+  - kaggle=1.5.12=py39hf3d152e_0
+  - keras-preprocessing=1.1.2=pyhd3eb1b0_0
+  - kiwisolver=1.3.1=py39h2531618_0
+  - krb5=1.17.1=h173b8e3_0
+  - lcms2=2.12=h3be6417_0
+  - ld_impl_linux-64=2.35.1=h7274673_9
+  - libclang=11.1.0=default_ha53f305_1
+  - libcurl=7.71.1=hcdd3856_8
+  - libedit=3.1.20210216=h27cfd23_1
+  - libev=4.33=h7b6447c_0
+  - libevent=2.1.10=hcdb4288_3
+  - libffi=3.3=he6710b0_2
+  - libgcc-ng=9.3.0=h5101ec6_17
+  - libgfortran-ng=7.5.0=ha8ba4b0_17
+  - libgfortran4=7.5.0=ha8ba4b0_17
+  - libglib=2.68.2=h3e27bee_0
+  - libgomp=9.3.0=h5101ec6_17
+  - libiconv=1.16=h516909a_0
+  - libllvm11=11.1.0=hf817b99_2
+  - libnghttp2=1.41.0=hf8bcb03_2
+  - libogg=1.3.5=h27cfd23_1
+  - libopus=1.3.1=h7b6447c_0
+  - libpng=1.6.37=hbc83047_0
+  - libpq=13.2=hfd2b0eb_2
+  - libprotobuf=3.14.0=h8c45485_0
+  - libssh2=1.9.0=h1ba5d50_1
+  - libstdcxx-ng=9.3.0=hd4cf53a_17
+  - libtiff=4.2.0=h85742a9_0
+  - libuuid=1.0.3=h1bed415_2
+  - libuv=1.40.0=h7b6447c_0
+  - libvorbis=1.3.7=h7b6447c_0
+  - libwebp-base=1.2.0=h27cfd23_0
+  - libxcb=1.14=h7b6447c_0
+  - libxkbcommon=1.0.3=he3ba5ed_0
+  - libxml2=2.9.12=h72842e0_0
+  - lz4-c=1.9.3=h2531618_0
+  - mako=1.1.4=pyhd3eb1b0_0
+  - markdown=3.3.4=py39h06a4308_0
+  - markupsafe=2.0.1=py39h27cfd23_0
+  - matplotlib=3.3.4=py39h06a4308_0
+  - matplotlib-base=3.3.4=py39h62a2d02_0
+  - mkl=2021.2.0=h06a4308_296
+  - mkl-service=2.3.0=py39h27cfd23_1
+  - mkl_fft=1.3.0=py39h42c9631_2
+  - mkl_random=1.2.1=py39ha9443f7_2
+  - mlflow=1.17.0=py39ha39b057_1
+  - multidict=5.1.0=py39h27cfd23_2
+  - munch=2.5.0=py_0
+  - mysql-common=8.0.25=ha770c72_0
+  - mysql-libs=8.0.25=h935591d_0
+  - ncurses=6.2=he6710b0_1
+  - ninja=1.10.2=hff7bd54_1
+  - nspr=4.30=h9c3ff4c_0
+  - nss=3.64=hb5efdd6_0
+  - numpy=1.20.2=py39h2d18471_0
+  - numpy-base=1.20.2=py39hfae3a4d_0
+  - oauthlib=3.1.1=pyhd3eb1b0_0
+  - olefile=0.46=py_0
+  - openssl=1.1.1k=h27cfd23_0
+  - opt_einsum=3.3.0=pyhd3eb1b0_1
+  - packaging=20.9=pyhd3eb1b0_0
+  - pandas=1.2.4=py39h2531618_0
+  - pcre=8.44=he6710b0_0
+  - pillow=8.2.0=py39he98fc37_0
+  - pip=21.1.2=py39h06a4308_0
+  - prometheus_client=0.11.0=pyhd3eb1b0_0
+  - prometheus_flask_exporter=0.18.2=pyhd8ed1ab_0
+  - protobuf=3.14.0=py39h2531618_1
+  - py-cpuinfo=8.0.0=pyhd3eb1b0_0
+  - pyasn1=0.4.8=py_0
+  - pyasn1-modules=0.2.8=py_0
+  - pycparser=2.20=py_2
+  - pyjwt=2.1.0=py39h06a4308_0
+  - pymongo=3.11.4=py39h2531618_0
+  - pyopenssl=20.0.1=pyhd3eb1b0_1
+  - pyparsing=2.4.7=pyhd3eb1b0_0
+  - pyqt=5.12.3=py39hf3d152e_7
+  - pyqt-impl=5.12.3=py39h0fcd23e_7
+  - pyqt5-sip=4.19.18=py39he80948d_7
+  - pyqtchart=5.12=py39h0fcd23e_7
+  - pyqtwebengine=5.12.1=py39h0fcd23e_7
+  - pysocks=1.7.1=py39h06a4308_0
+  - python=3.9.5=h12debd9_4
+  - python-dateutil=2.8.1=pyhd3eb1b0_0
+  - python-editor=1.0.4=py_0
+  - python-flatbuffers=1.12=pyhd3eb1b0_0
+  - python-slugify=5.0.2=pyhd3eb1b0_0
+  - python_abi=3.9=1_cp39
+  - pytorch=1.8.1=py3.9_cuda10.2_cudnn7.6.5_0
+  - pytz=2021.1=pyhd3eb1b0_0
+  - pyyaml=5.4.1=py39h27cfd23_1
+  - qt=5.12.9=hda022c4_4
+  - querystring_parser=1.2.4=py_0
+  - readline=8.1=h27cfd23_0
+  - requests=2.25.1=pyhd3eb1b0_0
+  - requests-oauthlib=1.3.0=py_0
+  - rsa=4.7.2=pyhd3eb1b0_1
+  - sacred=0.7.4=py_0
+  - scipy=1.6.2=py39had2a1c9_1
+  - seaborn=0.11.1=pyhd3eb1b0_0
+  - setuptools=52.0.0=py39h06a4308_0
+  - six=1.15.0=py39h06a4308_0
+  - smmap=4.0.0=pyhd3eb1b0_0
+  - snappy=1.1.8=he6710b0_0
+  - sqlalchemy=1.4.15=py39h27cfd23_0
+  - sqlite=3.35.4=hdfb4753_0
+  - sqlparse=0.4.1=py_0
+  - tabulate=0.8.9=py39h06a4308_0
+  - tensorboard=2.5.0=py_0
+  - tensorboard-plugin-wit=1.6.0=py_0
+  - tensorflow=2.5.0=mkl_py39h4a0693c_0
+  - tensorflow-base=2.5.0=mkl_py39h35b2a3d_0
+  - tensorflow-estimator=2.5.0=pyh7b7c402_0
+  - termcolor=1.1.0=py39h06a4308_1
+  - text-unidecode=1.3=py_0
+  - tk=8.6.10=hbc83047_0
+  - torchaudio=0.8.1=py39
+  - torchvision=0.2.2=py_3
+  - tornado=6.1=py39h27cfd23_0
+  - tqdm=4.59.0=pyhd3eb1b0_1
+  - typing-extensions=3.7.4.3=hd3eb1b0_0
+  - typing_extensions=3.7.4.3=pyh06a4308_0
+  - tzdata=2020f=h52ac0ba_0
+  - unidecode=1.2.0=pyhd3eb1b0_0
+  - urllib3=1.24.3=py_1
+  - websocket-client=0.58.0=py39h06a4308_4
+  - werkzeug=1.0.1=pyhd3eb1b0_0
+  - wheel=0.35.1=pyhd3eb1b0_0
+  - wrapt=1.12.1=py39he8ac12f_1
+  - xz=5.2.5=h7b6447c_0
+  - yaml=0.2.5=h7b6447c_0
+  - yarl=1.6.3=py39h27cfd23_0
+  - zipp=3.4.1=pyhd3eb1b0_0
+  - zlib=1.2.11=h7b6447c_3
+  - zstd=1.4.9=haebb681_0
+prefix: /home/filip/miniconda3/envs/iumenv