Skip to content

AryanSarswat/Early-Bird-Transformer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

To Reproduce The Experiments

Run the following command

python3 main.py

By Default the code will run CIFAR10 dataset with Vision Transformer model. To run other models and datasets, to run other datasets and models, you can change the parameters in the main.py file.

Replace the following line in the main.py file to run other models and datasets.

get_cifar10_dataloader()

with

get_cifar100_dataloader()

or

get_shvn_dataloader()

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published