Skip to content

Instantly share code, notes, and snippets.

View nikhil-salodkar's full-sized avatar

Nikhil Salodkar nikhil-salodkar

View GitHub Profile
@jbschiratti
jbschiratti / fine_tuning.py
Created April 8, 2020 09:29
Fine tuning on dummy data
"""Transfer learning example on fake data."""
from collections import OrderedDict
import torch
import torch.nn.functional as F
from torch import optim
from torch.utils.data import DataLoader
from torchvision.models import resnet50