• Nishant Prabhu's avatar
    Layoutlm onnx support (Issue #13300) (#13562) · ddd4d02f
    Nishant Prabhu authored
    
    
    * Add support for exporting PyTorch LayoutLM to ONNX
    
    * Added tests for converting LayoutLM to ONNX
    
    * Add support for exporting PyTorch LayoutLM to ONNX
    
    * Added tests for converting LayoutLM to ONNX
    
    * cleanup
    
    * Removed regression/ folder
    
    * Add support for exporting PyTorch LayoutLM to ONNX
    
    * Added tests for converting LayoutLM to ONNX
    
    * cleanup
    
    * Fixed import error
    
    * Remove unnecessary import statements
    
    * Changed max_2d_positions from class variable to instance variable of the config class
    
    * Add support for exporting PyTorch LayoutLM to ONNX
    
    * Added tests for converting LayoutLM to ONNX
    
    * cleanup
    
    * Add support for exporting PyTorch LayoutLM to ONNX
    
    * cleanup
    
    * Fixed import error
    
    * Changed max_2d_positions from class variable to instance variable of the config class
    
    * Use super class generate_dummy_inputs method
    Co-authored-by: default avatarMichael Benayoun <mickbenayoun@gmail.com>
    
    * Add support for Masked LM, sequence classification and token classification
    Co-authored-by: default avatarMichael Benayoun <mickbenayoun@gmail.com>
    
    * Removed uncessary import and method
    
    * Fixed code styling
    
    * Raise error if PyTorch is not installed
    
    * Remove unnecessary import statement
    Co-authored-by: default avatarMichael Benayoun <mickbenayoun@gmail.com>
    ddd4d02f
test_onnx_v2.py 11.7 KB