• Kamal Raj's avatar
    beit-flax (#13515) · a2dec768
    Kamal Raj authored
    * beit-flax
    
    * updated FLAX_BEIT_MLM_DOCSTRING
    
    * removed bool_masked_pos from classification
    
    * updated Copyright
    
    * code refactoring: x -> embeddings
    
    * updated test: rm from_pt
    
    * Update docs/source/model_doc/beit.rst
    
    * model code dtype updates and
    other changes according to review
    
    * relative_position_bias
    revert back to pytorch design
    a2dec768
beit.rst 6.47 KB