diff --git a/.openi-aitemplate b/.openi-aitemplate index 83e35d7..5886792 100644 --- a/.openi-aitemplate +++ b/.openi-aitemplate @@ -32,5 +32,44 @@ "BranchName": "liuzx", "BootFile": "gpu_mnist_example/train.py", "Parameters": [] + }, + { + "Name": "NPU训练任务", + "Description": "", + "JobType": "TRAIN", + "Cluster": "OpenI", + "ComputeSource": "NPU", + "HasInternet": 2, + "AccCardsNum": 1, + "AccCardType": "ASCEND910", + "CpuCores": 24, + "MemGiB": 256, + "GPUMemGiB": 32, + "ShareMemGiB": 0, + "Image": { + "ImageID": "-2", + "ImageName": "mindspore2.0-cann6", + "ImageUrl": "" + }, + "PretrainModelList": [ + { + "ID": "79942396-0f8d-4919-a490-a4b75a7da053", + "ModelName": "Mindspore_MNIST_Example_Model" + } + ], + "DatasetList": [ + { + "ID": "aced78db-6703-41b6-be46-140ccd411107", + "DatasetName": "MnistDataset_mindspore.zip" + } + ], + "BranchName": "master", + "BootFile": "npu_mnist_example/train.py", + "Parameters": [ + { + "Label": "epoch_size", + "Value": "5" + } + ] } ] \ No newline at end of file