GitHub - yongding-tao/Bootstrap_mae: The Bootstrapped version of MAE(The Coding Test of Tsinghua IIIS Professor YongGao)

Bootstrapped Masked Autoencoders

This repo is a modification on the MAE repo. Installation and preparation follow that repo.

You can run the shell in ./Experiment_Settints to conduct the experiment.

Like running :

sh ./Experiment_Settings/run_Bootstrap_MAE_onlyloadencoder_setting1.sh

Name		Name	Last commit message	Last commit date
Latest commit History 21 Commits
Experiment_Settings		Experiment_Settings
demo		demo
util		util
visualizer		visualizer
Bmae_eval_finetune.sh		Bmae_eval_finetune.sh
Bmae_eval_linear.sh		Bmae_eval_linear.sh
Bmae_train.sh		Bmae_train.sh
CODE_OF_CONDUCT.md		CODE_OF_CONDUCT.md
CONTRIBUTING.md		CONTRIBUTING.md
FINETUNE.md		FINETUNE.md
LICENSE		LICENSE
PRETRAIN.md		PRETRAIN.md
README.md		README.md
ema_mae_train.sh		ema_mae_train.sh
engine_finetune.py		engine_finetune.py
engine_pretrain.py		engine_pretrain.py
mae_eval_finetune.sh		mae_eval_finetune.sh
mae_eval_linear.sh		mae_eval_linear.sh
mae_train.sh		mae_train.sh
mae_train_normPixelLoss.sh		mae_train_normPixelLoss.sh
main_finetune.py		main_finetune.py
main_linprobe.py		main_linprobe.py
main_pretrain.py		main_pretrain.py
main_train_eval_noMAE.py		main_train_eval_noMAE.py
models_mae.py		models_mae.py
models_vit.py		models_vit.py
ne_Bmae_train.sh		ne_Bmae_train.sh
noMae_train_eval.sh		noMae_train_eval.sh
submitit_finetune.py		submitit_finetune.py
submitit_linprobe.py		submitit_linprobe.py
submitit_pretrain.py		submitit_pretrain.py