search

facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377
Other
6.93k stars 1.17k forks source link

Add torch hub support #127

Closed materight closed 1 year ago

materight commented 1 year ago

Allow to easily load the model from pytorch hub, e.g. using torch.hub.load('facebookresearch/mae:main', 'mae_vitb').