torch import

Solutions on MaxInterview for torch import by the best coders in the world

showing results for - "torch import"
Maëlys
17 Jul 2019
1import torch
2import torch.nn as nn
3import torch.nn.functional as F
4from torch.utils.data import DataLoader
5from torchvision import datasets, transforms, models
6from torchvision.utils import make_grid
7
8import numpy as np
9import pandas as pd
10from sklearn.metrics import confusion_matrix
11import matplotlib.pyplot as plt
12%matplotlib inline
13
14transform = transforms.ToTensor()
15
16train_data = datasets.FashionMNIST(root='../Data', train=True, download=True, transform=transform)
17test_data = datasets.FashionMNIST(root='../Data', train=False, download=True, transform=transform)
18