• Denisa Roberts's avatar
    Add TensorFlow implementation of EfficientFormer (#22620) · 88f50a1e
    Denisa Roberts authored
    * Add tf code for efficientformer
    
    * Fix return dict bug - return last hidden state after last stage
    
    * Fix corresponding return dict bug
    
    * Override test tol
    
    * Change default values of training to False
    
    * Set training to default False X3
    
    * Rm axis from ln
    
    * Set init in dense projection
    
    * Rm debug stuff
    
    * Make style; all tests pass.
    
    * Modify year to 2023
    
    * Fix attention biases codes
    
    * Update the shape list logic
    
    * Add a batch norm eps config
    
    * Remove extract comments in test files
    
    * Add conditional attn and hidden states return for serving output
    
    * Change channel dim checking logic
    
    * Add exception for withteacher model in training mode
    
    * Revert layer count for now
    
    * Add layer count for conditional layer naming
    
    * Transpose for conv happens only in main layer
    
    * Make tests smaller
    
    * Make style
    
    * Update doc
    
    * Rm from_pt
    
    * Change to actual expect image class label
    
    * Remove stray print in tests
    
    * Update image processor test
    
    * Remove the old serving output logic
    
    * Make style
    
    * Make style
    
    * Complete test
    88f50a1e
index.mdx 85.7 KB