Created
July 14, 2020 08:17
-
-
Save aletheia/5b370d04485932285693e0e1e53eb9b7 to your computer and use it in GitHub Desktop.
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
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