Merge pull request #5 from github/jungaretti/add-image-classifier

jungaretti · web-flow · commit a5f3fb7aaa17 · 2022-10-28T13:32:59.000-07:00
Add image classifier notebook
diff --git a/.gitignore b/.gitignore
@@ -0,0 +1,2 @@
+notebooks/data
+notebooks/cifar_net.pth
diff --git a/notebooks/image-classifier.ipynb b/notebooks/image-classifier.ipynb
diff --git a/requirements.txt b/requirements.txt
@@ -1,2 +1,7 @@
+ipywidgets==7.7.1
 matplotlib
+numpy
 pandas
+torch
+torchvision==0.13.1
+tqdm==4.64.0
