Skip to content

Instantly share code, notes, and snippets.

View crazyoscarchang's full-sized avatar

crazyoscarchang

View GitHub Profile
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import torchvision
from torchvision import datasets, transforms
import math
import numpy as np
# Hardcoded variables for hyperfan init