diff --git a/clearml/binding/frameworks/megengine_bind.py b/clearml/binding/frameworks/megengine_bind.py index 15fe422f..a06db8e5 100644 --- a/clearml/binding/frameworks/megengine_bind.py +++ b/clearml/binding/frameworks/megengine_bind.py @@ -1,6 +1,3 @@ -#!/usr/bin/env python3 -# -*- coding:utf-8 -*- - import sys import six @@ -15,6 +12,7 @@ from ...model import Framework class PatchMegEngineModelIO(PatchBaseModelIO): __main_task = None __patched = None + # __patched_lightning = None @staticmethod @@ -37,7 +35,7 @@ class PatchMegEngineModelIO(PatchBaseModelIO): # noinspection PyBroadException try: - import megengine as mge # noqa + import megengine as mge # noqa mge.save = _patched_call(mge.save, PatchMegEngineModelIO._save) mge.load = _patched_call(mge.load, PatchMegEngineModelIO._load) diff --git a/clearml/task.py b/clearml/task.py index 93fd14cf..4d7b2571 100644 --- a/clearml/task.py +++ b/clearml/task.py @@ -359,6 +359,7 @@ class Task(_Task): 'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True, 'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True, 'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True, + 'megengine': True, } :param bool auto_resource_monitoring: Automatically create machine resource monitoring plots diff --git a/examples/frameworks/megengine/megengine_mnist.py b/examples/frameworks/megengine/megengine_mnist.py index b1836d93..3f5f9d58 100644 --- a/examples/frameworks/megengine/megengine_mnist.py +++ b/examples/frameworks/megengine/megengine_mnist.py @@ -94,7 +94,7 @@ def train(dataloader, args): def main(): - task = Task.init(project_name='megengine', task_name='mge mnist train') # noqa + task = Task.init(project_name='examples', task_name='megengine mnist train') # noqa parser = argparse.ArgumentParser(description='MegEngine MNIST Example') parser.add_argument( diff --git a/examples/frameworks/megengine/requirements.txt b/examples/frameworks/megengine/requirements.txt new file mode 100644 index 00000000..e3c916c0 --- /dev/null +++ b/examples/frameworks/megengine/requirements.txt @@ -0,0 +1,3 @@ +MegEngine +tensorboardX +clearml \ No newline at end of file