@ -14,8 +14,8 @@ dependencies:
- pytorch=1.4.0
- torchvision=0.5
- tensorboard=2.1
- matplotlib
- scikit-learn
- matplotlib=3.1.3
- scikit-learn=0.22.1
- pyyaml=5.3.1
- nvidia-apex=0.1