Ziqun Ye

buildInitialTypesSimple throws TypeError: Cannot map this type to Onnx types: VectorUDT

2

Hi, I am trying to serialize the pyspark model RandomForestClassifier to onnx. First step, I tried to use buildInitialTypesSimple to get the intial types, but got this error. Any suggestions...

ValueError: Unable to create node 'TreeEnsembleClassifier' with name='TreeEnsembleClassifier'.

2

``` from sklearn.model_selection import train_test_split from sklearn.datasets import make_classification, make_regression import shutil import xgboost as xgb import onnx import onnxruntime as rt import numpy as np from sklearn.datasets import load_iris...

Save model checkpoint error when multi-gpu training still happens on 4.36.1

32

### System Info platform: linux python: 3.9 transformers: 4.36.1 running on two A10.2 ### Who can help? _No response_ ### Information - [ ] The official example scripts - [...