Skip to content

Commit 7b70f64

Browse files
authored
Update image_classification_example_mnist_cnn.py
1 parent 4e3484a commit 7b70f64

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

computer_vision/image_classification_example_mnist_cnn.py

+2
Original file line numberDiff line numberDiff line change
@@ -10,6 +10,8 @@
1010
@author: Dipu Kabir
1111
"""
1212

13+
! pip install torch
14+
1315
import torch
1416
import torchvision
1517
import torch.nn as nn

0 commit comments

Comments
 (0)