search

youqingxiaozhua / APViT

PaddlePaddle and PyTorch implementation of APViT and TransFER
Apache License 2.0
38 stars 10 forks source link

missing files #4

Closed firaskhemakhem closed 1 year ago

firaskhemakhem commented 1 year ago

I can't figure out where to get the 'ir.pdparams' and 'vit.pdparams' files in order to rebuild the model

youqingxiaozhua commented 1 year ago

Sorry for the confusion. These two param files are pre-trained weights for the IR-50 and ViT-Small models. You can download them following the Readme and convert them to paddle with the x2paddle tool. Notable, our paddle implementation is for inference only. Some crucial training details (e.g., data augmentations, learning rate scheduler) are not aligned with the paper.