ClearML/clearml
1
0
Fork 0
mirror of https://github.com/clearml/clearml synced 2025-06-26 18:16:07 +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
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -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
View File

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