Datasets
Models
Downloads: 1
Diff of
/src/main.py
[000000]
..
[a5c387]
Switch to side-by-side view
--- a +++ b/src/main.py @@ -0,0 +1,6 @@ +import torch +from torch.utils.data import Dataset +from torchvision import datasets +from torchvision.transforms import ToTensor +import matplotlib.pyplot as plt +