diff --git a/README.md b/README.md
index fd4cd4720314d7009ebf82d2bb5cfead1e8e9c5e..1c4a90f55a4e74dd24a8a4c14fc326dd2edffe4d 100644
--- a/README.md
+++ b/README.md
@@ -40,8 +40,8 @@ If way above does not work for you (e.g. you are on Windows), try the following
 ```bash
 conda create --name mpv-assignments-cpu-only python=3.6
 conda activate mpv-assignments-cpu-only
-conda install pytorch torchvision cpuonly -c pytorch
-pip install kornia tqdm notebook matplotlib opencv-contrib-python seaborn tensorboard tensorboardX
+conda install pytorch==1.4.0 torchvision==0.5.0 cpuonly -c pytorch
+pip install kornia==0.2.2 tqdm notebook matplotlib opencv-contrib-python seaborn tensorboard tensorboardX
 conda install -c conda-forge widgetsnbextension
 conda install -c conda-forge ipywidgets
 ```
@@ -51,8 +51,8 @@ And following for GPU:
 ```bash
 conda create --name mpv-assignments-gpu python=3.6
 conda activate mpv-assignments-gpu
-conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
-pip install kornia tqdm notebook matplotlib opencv-contrib-python seaborn tensorboard tensorboardX
+conda install pytorch==1.4.0 torchvision==0.5.0 cudatoolkit=10.1 -c pytorch
+pip install kornia==0.2.2 tqdm notebook matplotlib opencv-contrib-python seaborn tensorboard tensorboardX
 conda install -c conda-forge widgetsnbextension
 conda install -c conda-forge ipywidgets
 ```
diff --git a/conda_env_yaml/environment-cpu.yml b/conda_env_yaml/environment-cpu.yml
index 7a46428a341117d317bd7fb69eeb89f162c82cd2..a776454765332b87a622928fd7bc0aadcb1212af 100644
--- a/conda_env_yaml/environment-cpu.yml
+++ b/conda_env_yaml/environment-cpu.yml
@@ -4,123 +4,138 @@ channels:
   - conda-forge
   - defaults
 dependencies:
-  - _libgcc_mutex=0.1
-  - attrs=19.3.0
-  - backcall=0.1.0
-  - blas=1.0
-  - bleach=3.1.4
-  - ca-certificates=2020.4.5.1
-  - certifi=2020.4.5.1
-  - cpuonly=1.0
-  - decorator=4.4.2
-  - defusedxml=0.6.0
-  - entrypoints=0.3
-  - freetype=2.9.1
-  - importlib-metadata=1.6.0
-  - importlib_metadata=1.6.0
-  - intel-openmp=2020.0
-  - ipykernel=5.2.1
-  - ipython=7.13.0
-  - ipython_genutils=0.2.0
-  - ipywidgets=7.5.1
-  - jedi=0.17.0
-  - jinja2=2.11.2
-  - jpeg=9b
-  - jsonschema=3.2.0
-  - jupyter_client=6.1.3
-  - jupyter_core=4.6.3
-  - ld_impl_linux-64=2.33.1
-  - libedit=3.1.20181209
-  - libffi=3.2.1
-  - libgcc-ng=9.1.0
-  - libgfortran-ng=7.3.0
-  - libpng=1.6.37
-  - libsodium=1.0.17
-  - libstdcxx-ng=9.1.0
-  - libtiff=4.1.0
-  - markupsafe=1.1.1
-  - mistune=0.8.4
-  - mkl=2020.0
-  - mkl-service=2.3.0
-  - mkl_fft=1.0.15
-  - mkl_random=1.1.0
-  - nbconvert=5.6.1
-  - nbformat=5.0.6
-  - ncurses=6.2
-  - ninja=1.9.0
-  - notebook=6.0.3
-  - numpy=1.18.1
-  - numpy-base=1.18.1
-  - olefile=0.46
-  - openssl=1.1.1f
-  - pandoc=2.9.2.1
-  - parso=0.7.0
-  - pexpect=4.8.0
-  - pickleshare=0.7.5
-  - pillow=7.0.0
-  - pip=20.0.2
-  - prometheus_client=0.7.1
-  - prompt-toolkit=3.0.5
-  - ptyprocess=0.6.0
-  - pygments=2.6.1
-  - pyrsistent=0.16.0
-  - python=3.6.10
-  - python-dateutil=2.8.1
-  - python_abi=3.6
-  - pytorch=1.4.0
-  - readline=8.0
-  - send2trash=1.5.0
-  - setuptools=46.1.3
-  - six=1.14.0
-  - sqlite=3.31.1
-  - testpath=0.4.4
-  - tk=8.6.8
-  - torchvision=0.5.0
-  - tornado=6.0.4
-  - traitlets=4.3.3
-  - wcwidth=0.1.9
-  - wheel=0.34.2
-  - widgetsnbextension=3.5.1
-  - xz=5.2.5
-  - zeromq=4.3.2
-  - zipp=3.1.0
-  - zlib=1.2.11
-  - zstd=1.3.7
+  - _libgcc_mutex=0.1=main
+  - attrs=20.3.0=pyhd3deb0d_0
+  - backports=1.0=py_2
+  - backports.functools_lru_cache=1.6.1=py_0
+  - blas=1.0=mkl
+  - bleach=3.3.0=pyh44b312d_0
+  - ca-certificates=2020.12.5=ha878542_0
+  - certifi=2020.12.5=py36h5fab9bb_1
+  - cpuonly=1.0=0
+  - decorator=4.4.2=py_0
+  - defusedxml=0.7.1=pyhd8ed1ab_0
+  - entrypoints=0.3=pyhd8ed1ab_1003
+  - freetype=2.10.4=h5ab3b9f_0
+  - importlib-metadata=3.7.3=py36h5fab9bb_0
+  - intel-openmp=2020.2=254
+  - ipykernel=5.5.0=py36he448a4c_1
+  - ipython_genutils=0.2.0=py_1
+  - ipywidgets=7.6.3=pyhd3deb0d_0
+  - jinja2=2.11.3=pyh44b312d_0
+  - jpeg=9b=h024ee3a_2
+  - jsonschema=3.2.0=pyhd8ed1ab_3
+  - jupyter_client=6.1.12=pyhd8ed1ab_0
+  - jupyter_core=4.7.1=py36h5fab9bb_0
+  - jupyterlab_widgets=1.0.0=pyhd8ed1ab_1
+  - lcms2=2.11=h396b838_0
+  - ld_impl_linux-64=2.33.1=h53a641e_7
+  - libffi=3.3=he6710b0_2
+  - libgcc-ng=9.1.0=hdf63c60_0
+  - libpng=1.6.37=hbc83047_0
+  - libsodium=1.0.18=h36c2ea0_1
+  - libstdcxx-ng=9.1.0=hdf63c60_0
+  - libtiff=4.2.0=h85742a9_0
+  - libwebp-base=1.2.0=h27cfd23_0
+  - lz4-c=1.9.3=h2531618_0
+  - markupsafe=1.1.1=py36he6145b8_2
+  - mistune=0.8.4=py36h1d69622_1002
+  - mkl=2020.2=256
+  - mkl-service=2.3.0=py36he8ac12f_0
+  - mkl_fft=1.3.0=py36h54f3939_0
+  - mkl_random=1.1.1=py36h0573a6f_0
+  - nbformat=5.1.2=pyhd8ed1ab_1
+  - ncurses=6.2=he6710b0_1
+  - ninja=1.10.2=py36hff7bd54_0
+  - numpy=1.19.2=py36h54aff64_0
+  - numpy-base=1.19.2=py36hfa32c7d_0
+  - olefile=0.46=py36_0
+  - openssl=1.1.1j=h27cfd23_0
+  - packaging=20.9=pyh44b312d_0
+  - pandoc=2.12=h7f98852_0
+  - pexpect=4.8.0=pyh9f0ad1d_2
+  - pickleshare=0.7.5=py_1003
+  - pillow=8.1.2=py36he98fc37_0
+  - pip=21.0.1=py36h06a4308_0
+  - prometheus_client=0.9.0=pyhd3deb0d_0
+  - prompt_toolkit=1.0.15=py_1
+  - ptyprocess=0.7.0=pyhd3deb0d_0
+  - pygments=2.8.1=pyhd8ed1ab_0
+  - pyparsing=2.4.7=pyh9f0ad1d_0
+  - pyrsistent=0.17.3=py36h1d69622_1
+  - python=3.6.13=hdb3f193_0
+  - python-dateutil=2.8.1=py_0
+  - python_abi=3.6=1_cp36m
+  - pytorch=1.4.0=py3.6_cpu_0
+  - readline=8.1=h27cfd23_0
+  - send2trash=1.5.0=py_0
+  - setuptools=52.0.0=py36h06a4308_0
+  - simplegeneric=0.8.1=py_1
+  - six=1.15.0=py36h06a4308_0
+  - sqlite=3.35.2=hdfb4753_0
+  - terminado=0.9.3=py36h5fab9bb_0
+  - testpath=0.4.4=py_0
+  - tk=8.6.10=hbc83047_0
+  - torchvision=0.5.0=py36_cpu
+  - tornado=6.1=py36h1d69622_0
+  - traitlets=4.3.3=py36h9f0ad1d_1
+  - typing_extensions=3.7.4.3=py_0
+  - wcwidth=0.2.5=pyh9f0ad1d_2
+  - wheel=0.36.2=pyhd3eb1b0_0
+  - widgetsnbextension=3.5.1=py36h5fab9bb_4
+  - xz=5.2.5=h7b6447c_0
+  - zeromq=4.3.4=h2531618_0
+  - zipp=3.4.1=pyhd8ed1ab_0
+  - zlib=1.2.11=h7b6447c_3
+  - zstd=1.4.5=h9ceee32_0
   - pip:
-    - absl-py==0.9.0
-    - cachetools==4.1.0
-    - chardet==3.0.4
+    - absl-py==0.12.0
+    - argon2-cffi==20.1.0
+    - async-generator==1.10
+    - backcall==0.2.0
+    - cachetools==4.2.1
+    - cffi==1.14.5
+    - chardet==4.0.0
     - cycler==0.10.0
-    - google-auth==1.14.0
-    - google-auth-oauthlib==0.4.1
-    - grpcio==1.28.1
-    - idna==2.9
+    - google-auth==1.28.0
+    - google-auth-oauthlib==0.4.3
+    - grpcio==1.36.1
+    - idna==2.10
+    - ipython==7.16.1
     - ipython-genutils==0.2.0
-    - kiwisolver==1.2.0
-    - kornia==0.2.1
-    - markdown==3.2.1
-    - matplotlib==3.2.1
+    - jedi==0.18.0
+    - jupyterlab-pygments==0.1.2
+    - kiwisolver==1.3.1
+    - kornia==0.2.2
+    - markdown==3.3.4
+    - matplotlib==3.3.4
+    - nbclient==0.5.3
+    - nbconvert==6.0.7
+    - nest-asyncio==1.5.1
+    - notebook==6.3.0
     - oauthlib==3.1.0
-    - opencv-contrib-python==4.2.0.34
-    - pandas==1.0.3
-    - pandocfilters==1.4.2
-    - protobuf==3.11.3
+    - opencv-contrib-python==4.5.1.48
+    - pandas==1.1.5
+    - pandocfilters==1.4.3
+    - parso==0.8.1
+    - prompt-toolkit==3.0.18
+    - protobuf==3.15.6
     - pyasn1==0.4.8
     - pyasn1-modules==0.2.8
-    - pyparsing==2.4.7
-    - pytz==2019.3
-    - pyzmq==19.0.0
-    - requests==2.23.0
+    - pycparser==2.20
+    - pytz==2021.1
+    - pyzmq==22.0.3
+    - requests==2.25.1
     - requests-oauthlib==1.3.0
-    - rsa==4.0
-    - scipy==1.4.1
-    - seaborn==0.10.0
-    - tensorboard==2.2.1
-    - tensorboard-plugin-wit==1.6.0.post3
-    - tensorboardx==2.0
-    - terminado==0.8.3
-    - tqdm==4.45.0
-    - urllib3==1.25.9
+    - rsa==4.7.2
+    - scipy==1.5.4
+    - seaborn==0.11.1
+    - tensorboard==2.4.1
+    - tensorboard-plugin-wit==1.8.0
+    - tensorboardx==2.1
+    - tqdm==4.59.0
+    - urllib3==1.26.4
     - webencodings==0.5.1
     - werkzeug==1.0.1
+prefix: /home/old-ufo/anaconda3/envs/mpv-assignments-cpu-only
 
diff --git a/conda_env_yaml/environment-gpu.yml b/conda_env_yaml/environment-gpu.yml
index 295b618a69424502d9fd1726b722bc0a5e5ed760..e1ca0bc1168b5620ba1f7572ef456a9e855bdd09 100644
--- a/conda_env_yaml/environment-gpu.yml
+++ b/conda_env_yaml/environment-gpu.yml
@@ -4,123 +4,138 @@ channels:
   - conda-forge
   - defaults
 dependencies:
-  - _libgcc_mutex=0.1
-  - attrs=19.3.0
-  - backcall=0.1.0
-  - blas=1.0
-  - bleach=3.1.4
-  - ca-certificates=2020.4.5.1
-  - certifi=2020.4.5.1
-  - cudatoolkit=10.1.243
-  - decorator=4.4.2
-  - defusedxml=0.6.0
-  - entrypoints=0.3
-  - freetype=2.9.1
-  - importlib-metadata=1.6.0
-  - importlib_metadata=1.6.0
-  - intel-openmp=2020.0
-  - ipykernel=5.2.1
-  - ipython=7.13.0
-  - ipython_genutils=0.2.0
-  - ipywidgets=7.5.1
-  - jedi=0.17.0
-  - jinja2=2.11.2
-  - jpeg=9b
-  - jsonschema=3.2.0
-  - jupyter_client=6.1.3
-  - jupyter_core=4.6.3
-  - ld_impl_linux-64=2.33.1
-  - libedit=3.1.20181209
-  - libffi=3.2.1
-  - libgcc-ng=9.1.0
-  - libgfortran-ng=7.3.0
-  - libpng=1.6.37
-  - libsodium=1.0.17
-  - libstdcxx-ng=9.1.0
-  - libtiff=4.1.0
-  - markupsafe=1.1.1
-  - mistune=0.8.4
-  - mkl=2020.0
-  - mkl-service=2.3.0
-  - mkl_fft=1.0.15
-  - mkl_random=1.1.0
-  - nbconvert=5.6.1
-  - nbformat=5.0.6
-  - ncurses=6.2
-  - ninja=1.9.0
-  - notebook=6.0.3
-  - numpy=1.18.1
-  - numpy-base=1.18.1
-  - olefile=0.46
-  - openssl=1.1.1g
-  - pandoc=2.9.2.1
-  - parso=0.7.0
-  - pexpect=4.8.0
-  - pickleshare=0.7.5
-  - pillow=7.0.0
-  - pip=20.0.2
-  - prometheus_client=0.7.1
-  - prompt-toolkit=3.0.5
-  - ptyprocess=0.6.0
-  - pygments=2.6.1
-  - pyrsistent=0.16.0
-  - python=3.6.10
-  - python-dateutil=2.8.1
-  - python_abi=3.6
-  - pytorch=1.4.0
-  - readline=8.0
-  - send2trash=1.5.0
-  - setuptools=46.1.3
-  - six=1.14.0
-  - sqlite=3.31.1
-  - testpath=0.4.4
-  - tk=8.6.8
-  - torchvision=0.5.0
-  - tornado=6.0.4
-  - traitlets=4.3.3
-  - wcwidth=0.1.9
-  - wheel=0.34.2
-  - widgetsnbextension=3.5.1
-  - xz=5.2.5
-  - zeromq=4.3.2
-  - zipp=3.1.0
-  - zlib=1.2.11
-  - zstd=1.3.7
+  - _libgcc_mutex=0.1=main
+  - attrs=20.3.0=pyhd3deb0d_0
+  - backports=1.0=py_2
+  - backports.functools_lru_cache=1.6.1=py_0
+  - blas=1.0=mkl
+  - bleach=3.3.0=pyh44b312d_0
+  - ca-certificates=2020.12.5=ha878542_0
+  - certifi=2020.12.5=py36h5fab9bb_1
+  - cudatoolkit=10.1.243=h6bb024c_0
+  - decorator=4.4.2=py_0
+  - defusedxml=0.7.1=pyhd8ed1ab_0
+  - entrypoints=0.3=pyhd8ed1ab_1003
+  - freetype=2.10.4=h5ab3b9f_0
+  - importlib-metadata=3.7.3=py36h5fab9bb_0
+  - intel-openmp=2020.2=254
+  - ipykernel=5.5.0=py36he448a4c_1
+  - ipython_genutils=0.2.0=py_1
+  - ipywidgets=7.6.3=pyhd3deb0d_0
+  - jinja2=2.11.3=pyh44b312d_0
+  - jpeg=9b=h024ee3a_2
+  - jsonschema=3.2.0=pyhd8ed1ab_3
+  - jupyter_client=6.1.12=pyhd8ed1ab_0
+  - jupyter_core=4.7.1=py36h5fab9bb_0
+  - jupyterlab_widgets=1.0.0=pyhd8ed1ab_1
+  - lcms2=2.11=h396b838_0
+  - ld_impl_linux-64=2.33.1=h53a641e_7
+  - libffi=3.3=he6710b0_2
+  - libgcc-ng=9.1.0=hdf63c60_0
+  - libpng=1.6.37=hbc83047_0
+  - libsodium=1.0.18=h36c2ea0_1
+  - libstdcxx-ng=9.1.0=hdf63c60_0
+  - libtiff=4.2.0=h85742a9_0
+  - libwebp-base=1.2.0=h27cfd23_0
+  - lz4-c=1.9.3=h2531618_0
+  - markupsafe=1.1.1=py36he6145b8_2
+  - mistune=0.8.4=py36h1d69622_1002
+  - mkl=2020.2=256
+  - mkl-service=2.3.0=py36he8ac12f_0
+  - mkl_fft=1.3.0=py36h54f3939_0
+  - mkl_random=1.1.1=py36h0573a6f_0
+  - nbformat=5.1.2=pyhd8ed1ab_1
+  - ncurses=6.2=he6710b0_1
+  - ninja=1.10.2=py36hff7bd54_0
+  - numpy=1.19.2=py36h54aff64_0
+  - numpy-base=1.19.2=py36hfa32c7d_0
+  - olefile=0.46=py36_0
+  - openssl=1.1.1j=h27cfd23_0
+  - packaging=20.9=pyh44b312d_0
+  - pandoc=2.12=h7f98852_0
+  - pexpect=4.8.0=pyh9f0ad1d_2
+  - pickleshare=0.7.5=py_1003
+  - pillow=8.1.2=py36he98fc37_0
+  - pip=21.0.1=py36h06a4308_0
+  - prometheus_client=0.9.0=pyhd3deb0d_0
+  - prompt_toolkit=1.0.15=py_1
+  - ptyprocess=0.7.0=pyhd3deb0d_0
+  - pygments=2.8.1=pyhd8ed1ab_0
+  - pyparsing=2.4.7=pyh9f0ad1d_0
+  - pyrsistent=0.17.3=py36h1d69622_1
+  - python=3.6.13=hdb3f193_0
+  - python-dateutil=2.8.1=py_0
+  - python_abi=3.6=1_cp36m
+  - pytorch=1.4.0=py3.6_cuda10.1.243_cudnn7.6.3_0
+  - readline=8.1=h27cfd23_0
+  - send2trash=1.5.0=py_0
+  - setuptools=52.0.0=py36h06a4308_0
+  - simplegeneric=0.8.1=py_1
+  - six=1.15.0=py36h06a4308_0
+  - sqlite=3.35.2=hdfb4753_0
+  - terminado=0.9.3=py36h5fab9bb_0
+  - testpath=0.4.4=py_0
+  - tk=8.6.10=hbc83047_0
+  - torchvision=0.5.0=py36_cu101
+  - tornado=6.1=py36h1d69622_0
+  - traitlets=4.3.3=py36h9f0ad1d_1
+  - typing_extensions=3.7.4.3=py_0
+  - wcwidth=0.2.5=pyh9f0ad1d_2
+  - wheel=0.36.2=pyhd3eb1b0_0
+  - widgetsnbextension=3.5.1=py36h5fab9bb_4
+  - xz=5.2.5=h7b6447c_0
+  - zeromq=4.3.4=h2531618_0
+  - zipp=3.4.1=pyhd8ed1ab_0
+  - zlib=1.2.11=h7b6447c_3
+  - zstd=1.4.5=h9ceee32_0
   - pip:
-    - absl-py==0.9.0
-    - cachetools==4.1.0
-    - chardet==3.0.4
+    - absl-py==0.12.0
+    - argon2-cffi==20.1.0
+    - async-generator==1.10
+    - backcall==0.2.0
+    - cachetools==4.2.1
+    - cffi==1.14.5
+    - chardet==4.0.0
     - cycler==0.10.0
-    - google-auth==1.14.0
-    - google-auth-oauthlib==0.4.1
-    - grpcio==1.28.1
-    - idna==2.9
+    - google-auth==1.28.0
+    - google-auth-oauthlib==0.4.3
+    - grpcio==1.36.1
+    - idna==2.10
+    - ipython==7.16.1
     - ipython-genutils==0.2.0
-    - kiwisolver==1.2.0
-    - kornia==0.2.1
-    - markdown==3.2.1
-    - matplotlib==3.2.1
+    - jedi==0.18.0
+    - jupyterlab-pygments==0.1.2
+    - kiwisolver==1.3.1
+    - kornia==0.2.2
+    - markdown==3.3.4
+    - matplotlib==3.3.4
+    - nbclient==0.5.3
+    - nbconvert==6.0.7
+    - nest-asyncio==1.5.1
+    - notebook==6.3.0
     - oauthlib==3.1.0
-    - opencv-contrib-python==4.2.0.34
-    - pandas==1.0.3
-    - pandocfilters==1.4.2
-    - protobuf==3.11.3
+    - opencv-contrib-python==4.5.1.48
+    - pandas==1.1.5
+    - pandocfilters==1.4.3
+    - parso==0.8.1
+    - prompt-toolkit==3.0.18
+    - protobuf==3.15.6
     - pyasn1==0.4.8
     - pyasn1-modules==0.2.8
-    - pyparsing==2.4.7
-    - pytz==2019.3
-    - pyzmq==19.0.0
-    - requests==2.23.0
+    - pycparser==2.20
+    - pytz==2021.1
+    - pyzmq==22.0.3
+    - requests==2.25.1
     - requests-oauthlib==1.3.0
-    - rsa==4.0
-    - scipy==1.4.1
-    - seaborn==0.10.0
-    - tensorboard==2.2.1
-    - tensorboard-plugin-wit==1.6.0.post3
-    - tensorboardx==2.0
-    - terminado==0.8.3
-    - tqdm==4.45.0
-    - urllib3==1.25.9
+    - rsa==4.7.2
+    - scipy==1.5.4
+    - seaborn==0.11.1
+    - tensorboard==2.4.1
+    - tensorboard-plugin-wit==1.8.0
+    - tensorboardx==2.1
+    - tqdm==4.59.0
+    - urllib3==1.26.4
     - webencodings==0.5.1
     - werkzeug==1.0.1
+prefix: /home/old-ufo/anaconda3/envs/mpv-assignments-gpu