Skip to content

Instantly share code, notes, and snippets.

View t-vi's full-sized avatar
🌍
Creating fun and friendly AIs

Thomas Viehmann t-vi

🌍
Creating fun and friendly AIs
View GitHub Profile
import torch
from PIL import Image
from torch.utils.data import DataLoader
import torchvision
from torchvision import transforms, datasets
from torch.autograd import Variable
import torch.nn as nn
import torch.optim
import torch.backends.cudnn as cudnn; cudnn.benchmark = True