ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-03-03 10:42:00 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add MegEngine example requirements and set examples project

Browse Source
This commit is contained in:
allegroai 2021-09-30 22:23:23 +03:00
parent d0e4e0be64
commit b38a74c91e
4 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
clearml/binding/frameworks/megengine_bind.py
View File

@ -1,6 +1,3 @@
#!/usr/bin/env python3
# -*- coding:utf-8 -*-
import sys import sys
import six import six
@ -15,6 +12,7 @@ from ...model import Framework
class PatchMegEngineModelIO(PatchBaseModelIO): class PatchMegEngineModelIO(PatchBaseModelIO):
__main_task = None __main_task = None
__patched = None __patched = None
# __patched_lightning = None # __patched_lightning = None
@staticmethod @staticmethod
@ -37,7 +35,7 @@ class PatchMegEngineModelIO(PatchBaseModelIO):
# noinspection PyBroadException # noinspection PyBroadException
try: try:
import megengine as mge # noqa import megengine as mge # noqa
mge.save = _patched_call(mge.save, PatchMegEngineModelIO._save) mge.save = _patched_call(mge.save, PatchMegEngineModelIO._save)
mge.load = _patched_call(mge.load, PatchMegEngineModelIO._load) mge.load = _patched_call(mge.load, PatchMegEngineModelIO._load)

1
clearml/task.py
View File

@ -359,6 +359,7 @@ class Task(_Task):
'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True, 'matplotlib': True, 'tensorflow': True, 'tensorboard': True, 'pytorch': True,
'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True, 'xgboost': True, 'scikit': True, 'fastai': True, 'lightgbm': True,
'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True, 'hydra': True, 'detect_repository': True, 'tfdefines': True, 'joblib': True,
'megengine': True,
} }
:param bool auto_resource_monitoring: Automatically create machine resource monitoring plots :param bool auto_resource_monitoring: Automatically create machine resource monitoring plots

2
examples/frameworks/megengine/megengine_mnist.py
View File

@ -94,7 +94,7 @@ def train(dataloader, args):
def main(): 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 = argparse.ArgumentParser(description='MegEngine MNIST Example')
parser.add_argument( parser.add_argument(

3
examples/frameworks/megengine/requirements.txt Normal file
View File

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