View Code? Open in Web Editor
NEW
This project forked from tuyenldhust/mae-aflw
Shell 0.05%
Python 5.10%
Jupyter Notebook 94.86%
mae-aflw's Introduction
1. Use dataset croped face 64x64 on Kaggle
2. Config WANDB_API_KEY on configs/pretrain.yaml or configs/finetune.yaml and config other hyperparameters
- Get API key on WANDB
- Config other hyperparameters in configs/pretrain.yaml or configs/finetune.yaml
- Run command:
python train.py --cfg configs/pretrain.yaml
- Run command:
python finetune.py --cfg configs/finetune.yaml --pretrained <path_to_pretrain_model>
- Resume finetune from checkpoint file:
python finetune.py --cfg configs/finetune.yaml --resume <path_to_checkpoint>
![image](https://private-user-images.githubusercontent.com/19906050/246727558-694769ce-f7aa-4b55-88d2-14b06771dd36.png?jwt=eyJhbGciOiJIUzI1NiIsInR5cCI6IkpXVCJ9.eyJpc3MiOiJnaXRodWIuY29tIiwiYXVkIjoicmF3LmdpdGh1YnVzZXJjb250ZW50LmNvbSIsImtleSI6ImtleTUiLCJleHAiOjE3MTkzOTA3ODUsIm5iZiI6MTcxOTM5MDQ4NSwicGF0aCI6Ii8xOTkwNjA1MC8yNDY3Mjc1NTgtNjk0NzY5Y2UtZjdhYS00YjU1LTg4ZDItMTRiMDY3NzFkZDM2LnBuZz9YLUFtei1BbGdvcml0aG09QVdTNC1ITUFDLVNIQTI1NiZYLUFtei1DcmVkZW50aWFsPUFLSUFWQ09EWUxTQTUzUFFLNFpBJTJGMjAyNDA2MjYlMkZ1cy1lYXN0LTElMkZzMyUyRmF3czRfcmVxdWVzdCZYLUFtei1EYXRlPTIwMjQwNjI2VDA4MjgwNVomWC1BbXotRXhwaXJlcz0zMDAmWC1BbXotU2lnbmF0dXJlPTE0YjE5M2FlYjI4ZDg0MGQzYjNlMWFmNDhhNjE3OThmNDNkZmQyMzIyN2NjN2VhODc0NmVkY2RhMzA3NjI1MDgmWC1BbXotU2lnbmVkSGVhZGVycz1ob3N0JmFjdG9yX2lkPTAma2V5X2lkPTAmcmVwb19pZD0wIn0.Bvr57jiNX9B1OWP6A3hM4NWbExJycctY8D28H9s3sjk)
mae-aflw's People
Contributors