Datasets
Models
Downloads: 1
Diff of
/src/main.py
[000000]
..
[a5c387]
Switch to unified view
| a | b/src/main.py | ||
|---|---|---|---|
| 1 | import torch |
||
| 2 | from torch.utils.data import Dataset |
||
| 3 | from torchvision import datasets |
||
| 4 | from torchvision.transforms import ToTensor |
||
| 5 | import matplotlib.pyplot as plt |
||
| 6 |