Skip to content

Instantly share code, notes, and snippets.

View lewisjiang's full-sized avatar

Bratislava lewisjiang

  • Hong Kong
View GitHub Profile
import argparse
import torch
import torch.nn as nn
from torch.autograd import Variable
from torch.utils.data import DataLoader
import torchvision
import torchvision.transforms as T
from torchvision.datasets import ImageFolder