Skip to content

Instantly share code, notes, and snippets.

View comeeasy's full-sized avatar

joono comeeasy

View GitHub Profile
@comeeasy
comeeasy / vit.py
Created September 26, 2024 05:32
ViT example
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
from torchvision import datasets, transforms
from torch.utils.data import DataLoader, Subset
import random