KimRass / MAE Public

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

PyTorch implementation of 'MAE' (He et al., 2021) from scratch.

1 star 0 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 10 Commits
papers		papers
samples		samples
.gitignore		.gitignore
README.md		README.md
model.py		model.py
pseudo_train.py		pseudo_train.py
set_pythonpath.sh		set_pythonpath.sh
utils.py		utils.py

Repository files navigation

Samples from Pseudo-pre-trained Model

Masked image, reconstructed image and original image

2. To-Dos

[✓] Architecture for pre-training
[✓] Architecture for self-supervised learning
[✗] Training on ImageNet-1K
[✗] Fine-tuning
[✗] Linear probing

3. References

[1] https://github.com/IcarusWizard/MAE

About

PyTorch implementation of 'MAE' (He et al., 2021) from scratch.

Report repository

Releases

No releases published

Packages

No packages published

Languages