This repository has been archived by the owner on Jul 10, 2024. It is now read-only.
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
SUBMARINE-678. add pytorch template to mysql
### What is this PR for? The only predefined template currently in the database is tf-mnist I add the pytorch template into the database The following is the content ` { "name": "pytorch-mnist", "author": "author", "description": "This is a template to run pytorch-mnist\n", "parameters": [ { "name": "experiment_name", "required": "true", "description": "experiment name", "value": null } ], "experimentSpec": { "meta": { "name": "{{experiment_name}}", "namespace": "default", "framework": "PyTorch", "cmd": "python /var/mnist.py --backend gloo", "envVars": { "ENV_1": "ENV1" } }, "environment": { "image": "apache/submarine:pytorch-dist-mnist-1.0" }, "spec": { "Master": { "replicas": 1, "resources": "cpu=1,memory=1024M" }, "Worker": { "replicas": 1, "resources": "cpu=1,memory=1024M" } } } } ` ### What type of PR is it? Improvement ### Todos * [O] - add pytorch-mnist to database ### What is the Jira issue? https://issues.apache.org/jira/projects/SUBMARINE/issues/SUBMARINE-678 ### How should this be tested? ` curl -X POST -H "Content-Type: application/json" -d ' { "params": { "experiment_name":"new-pytorch-mnist" } } ' http://127.0.0.1:8080/api/v1/experiment/pytorch-mnist ` ### Screenshots (if appropriate) ![image](https://user-images.githubusercontent.com/19265751/100550083-6d0bf600-32b2-11eb-87f2-95d7aadd66a9.png) ### Questions: * Does the licenses files need update? No * Is there breaking changes for older versions? No * Does this needs documentation? No Author: JohnTing <[email protected]> Closes #466 from JohnTing/SUBMARINE-678 and squashes the following commits: 4c173b3 [JohnTing] add doc 65c0dd1 [JohnTing] Fix typo: pytprch->pytorch 9946e32 [JohnTing] add pytprch-mnist template to database
- Loading branch information