Add MegEngine example requirements and set examples project

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)
 

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

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(

examples/frameworks/megengine/requirements.txt

@@ -0,0 +1,3 @@
+MegEngine
+tensorboardX
+clearml