From 9f1c63ca15d76aaec566cb33351f1c496cf3d735 Mon Sep 17 00:00:00 2001 From: pollfly <75068813+pollfly@users.noreply.github.com> Date: Sun, 20 Mar 2022 12:55:27 +0200 Subject: [PATCH] Update supported frameworks list (#209) --- docs/faq.md | 1 + docs/fundamentals/artifacts.md | 13 ++++++++++--- docs/integrations/libraries.md | 14 +++++++------- 3 files changed, 18 insertions(+), 10 deletions(-) diff --git a/docs/faq.md b/docs/faq.md index 22c33b8a..2d474658 100644 --- a/docs/faq.md +++ b/docs/faq.md @@ -470,6 +470,7 @@ auto_connect_frameworks={ 'matplotlib': True, 'tensorflow': False, 'tensorboard': False, 'pytorch': True, 'xgboost': False, 'scikit': True, 'fastai': True, 'lightgbm': False, 'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True, + 'megengine': True, 'jsonargparse': True, 'catboost': True } ``` diff --git a/docs/fundamentals/artifacts.md b/docs/fundamentals/artifacts.md index f94b4129..6200e4f5 100644 --- a/docs/fundamentals/artifacts.md +++ b/docs/fundamentals/artifacts.md @@ -163,8 +163,15 @@ This is especially helpful when using [clearml-agent](../clearml_agent.md) to ex ### List of Supported Frameworks -- Tensorflow +- TensorFlow - Keras -- Pytorch +- PyTorch +- PyTorch Ignite +- PyTorch Lightning - scikit-learn (only using joblib) -- XGBoost (only using joblib) \ No newline at end of file +- XGBoost (only using joblib) +- AutoKeras +- FastAI +- LightGBM +- MegEngine +- CatBoost \ No newline at end of file diff --git a/docs/integrations/libraries.md b/docs/integrations/libraries.md index 9411b8e2..894469a6 100644 --- a/docs/integrations/libraries.md +++ b/docs/integrations/libraries.md @@ -9,19 +9,19 @@ Just follow the [getting started](/getting_started/ds/ds_first_steps.md) to auto ![Integration tools](../img/integration_tools.png) **Frameworks** -- [Pytorch](https://github.com/allegroai/clearml/tree/master/examples/frameworks/pytorch) -- [pytorch-lightning](https://github.com/allegroai/clearml/tree/master/examples/frameworks/pytorch-lightning) -- [ignite](https://github.com/allegroai/clearml/tree/master/examples/frameworks/ignite) -- [Tensorflow](https://github.com/allegroai/clearml/tree/master/examples/frameworks/tensorflow) +- [PyTorch](https://github.com/allegroai/clearml/tree/master/examples/frameworks/pytorch) +- [PyTorch Lightning](https://github.com/allegroai/clearml/tree/master/examples/frameworks/pytorch-lightning) +- [PyTorch Ignite](https://github.com/allegroai/clearml/tree/master/examples/frameworks/ignite) +- [TensorFlow](https://github.com/allegroai/clearml/tree/master/examples/frameworks/tensorflow) - [Keras](https://github.com/allegroai/clearml/tree/master/examples/frameworks/keras) - [scikit-learn](https://github.com/allegroai/clearml/tree/master/examples/frameworks/scikit-learn) -- [fastai](https://github.com/allegroai/clearml/tree/master/examples/frameworks/fastai) -- [lightgbm](https://github.com/allegroai/clearml/tree/master/examples/frameworks/lightgbm) +- [FastAI](https://github.com/allegroai/clearml/tree/master/examples/frameworks/fastai) +- [LightGBM](https://github.com/allegroai/clearml/tree/master/examples/frameworks/lightgbm) - [XGBoost](https://github.com/allegroai/clearml/tree/master/examples/frameworks/xgboost) - [MegEngine](https://github.com/allegroai/clearml/tree/master/examples/frameworks/megengine) - [CatBoost](https://github.com/allegroai/clearml/tree/master/examples/frameworks/catboost) - [Hydra](https://github.com/allegroai/clearml/tree/master/examples/frameworks/hydra) -- [python-fire](https://github.com/allegroai/clearml/tree/master/examples/frameworks/fire) +- [Python Fire](https://github.com/allegroai/clearml/tree/master/examples/frameworks/fire) - [click](https://github.com/allegroai/clearml/tree/master/examples/frameworks/click) **HPO**