ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-02-07 13:23:40 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add more model frameworks to the list

Browse Source
This commit is contained in:
allegroai 2022-08-20 22:57:20 +03:00
parent 5a83aa433d
commit ca384aa75c
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
clearml/model.py
View File

@ -40,6 +40,7 @@ class Framework(Options):
tensorflowjs = 'TensorFlow_js' tensorflowjs = 'TensorFlow_js'
tensorflowlite = 'TensorFlow_Lite' tensorflowlite = 'TensorFlow_Lite'
pytorch = 'PyTorch' pytorch = 'PyTorch'
torchscript = 'TorchScript'
caffe = 'Caffe' caffe = 'Caffe'
caffe2 = 'Caffe2' caffe2 = 'Caffe2'
onnx = 'ONNX' onnx = 'ONNX'
@ -55,6 +56,8 @@ class Framework(Options):
parquet = 'Parquet' parquet = 'Parquet'
megengine = 'MegEngine' megengine = 'MegEngine'
catboost = 'CatBoost' catboost = 'CatBoost'
tensorrt = 'TensorRT'
openvino = 'OpenVINO'
__file_extensions_mapping = { __file_extensions_mapping = {
'.pb': (tensorflow, tensorflowjs, onnx, ), '.pb': (tensorflow, tensorflowjs, onnx, ),
@ -84,6 +87,7 @@ class Framework(Options):
'.pkl': (scikitlearn, keras, xgboost, megengine), '.pkl': (scikitlearn, keras, xgboost, megengine),
'.parquet': (parquet, ), '.parquet': (parquet, ),
'.cbm': (catboost, ), '.cbm': (catboost, ),
'.plan': (tensorrt,),
} }
__parent_mapping = { __parent_mapping = {