lsqqqq / jigsaw_vit Public

Notifications You must be signed in to change notification settings
Fork 0
Star 2

The reproduction of paper Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzles

2 stars 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
model		model
scripts		scripts
util		util
.gitignore		.gitignore
README.md		README.md
load_vit_from_ckpt.py		load_vit_from_ckpt.py
permutations.npy		permutations.npy
select_permutations.py		select_permutations.py
train.py		train.py

Repository files navigation

[Reproduce] Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzles (ViT)

ECCV 2016: Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzles

Reproduced based on https://github.com/YangZyyyy/JigsawPuzzlesPytorch

Changes:

Replaced AlexNet with a Vision Transformer
Enabled DDP and AMP

Environment:

python 3.9
pytorch 2.0.0

Generate permutation

python select_permutation.py

Train

cd scripts
bash run.sh

About

The reproduction of paper Unsupervised Learning of Visual Representations by Solving Jigsaw Puzzles

Report repository

Releases

No releases published

Packages

No packages published

Languages