Skip to content

Instantly share code, notes, and snippets.

@aletheia
Created July 14, 2020 08:17
Show Gist options
  • Save aletheia/5b370d04485932285693e0e1e53eb9b7 to your computer and use it in GitHub Desktop.
Save aletheia/5b370d04485932285693e0e1e53eb9b7 to your computer and use it in GitHub Desktop.
import os
import math
import random as rn
import numpy as np
import torch
import torch.nn as nn
from torch.nn import functional as F
from torch.utils.data import DataLoader
from torch.utils.data.sampler import SubsetRandomSampler
from torchvision import transforms as T, datasets
import pytorch_lightning as pl
class MNISTClassifier(pl.LightningModule):
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment