Skip to content

Instantly share code, notes, and snippets.

Avatar

Debaleena Misra d-misra

  • Grenoble
View GitHub Profile
View pytorch_finetune.py
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