mirror of
https://github.com/clearml/clearml
synced 2025-03-03 10:42:00 +00:00
Add MegEngine example requirements and set examples project
This commit is contained in:
parent
d0e4e0be64
commit
b38a74c91e
@ -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
|
||||
|
@ -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
|
||||
|
@ -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(
|
||||
|
3
examples/frameworks/megengine/requirements.txt
Normal file
3
examples/frameworks/megengine/requirements.txt
Normal file
@ -0,0 +1,3 @@
|
||||
MegEngine
|
||||
tensorboardX
|
||||
clearml
|
Loading…
Reference in New Issue
Block a user