• NielsRogge's avatar
    Add MAE (#15120) · 74bec986
    NielsRogge authored
    * First draft
    
    * More improvements
    
    * More improvements
    
    * More improvements
    
    * Fix embeddings
    
    * Add conversion script
    
    * Finish conversion script
    
    * More improvements
    
    * Fix forward pass
    
    * Remove print statements
    
    * Add weights initialization
    
    * Add initialization of decoder weights
    
    * Add support for other models in the conversion script
    
    * Fix patch_size for huge model
    
    * Fix most of the tests
    
    * Fix integration test
    
    * Fix docs
    
    * Fix archive_list
    
    * Apply suggestions from code review
    
    * Improve documentation
    
    * Apply more suggestions
    
    * Skip some tests due to non-deterministic behaviour
    
    * Fix test_initialization
    
    * Remove unneccessary initialization of nn.Embedding
    
    * Improve docs
    
    * Fix dummies
    
    * Remove ViTMAEFeatureExtractor from docs
    
    * Add model to README and table of contents
    
    * Delete inference file
    74bec986
dummy_pt_objects.py 92.3 KB