From 1f459ba19422593de325983040e176f97cf4ffc0 Mon Sep 17 00:00:00 2001
From: aktersnurra <gustaf.rydholm@gmail.com>
Date: Thu, 20 Aug 2020 22:18:35 +0200
Subject: A lot of stuff working :D. ResNet implemented!

---
 pyproject.toml | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'pyproject.toml')

diff --git a/pyproject.toml b/pyproject.toml
index 15d1f57..49bb049 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -22,8 +22,8 @@ sphinx_rtd_theme = "^0.4.3"
 boltons = "^20.1.0"
 h5py = "^2.10.0"
 toml = "^0.10.1"
-torch = "^1.5.0"
-torchvision = "^0.6.0"
+torch = "^1.6.0"
+torchvision = "^0.7.0"
 torchsummary = "^1.5.1"
 loguru = "^0.5.0"
 matplotlib = "^3.2.1"
@@ -32,6 +32,7 @@ pytest = "^5.4.3"
 opencv-python = "^4.3.0"
 nltk = "^3.5"
 einops = "^0.2.0"
+wandb = "^0.9.5"
 
 [tool.poetry.dev-dependencies]
 pytest = "^5.4.2"
-- 
cgit v1.2.3-70-g09d2