From d9f95dd927b733f71dd1aaaf558f4f93fea93c34 Mon Sep 17 00:00:00 2001 From: liuzx Date: Thu, 27 Jun 2024 10:14:18 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20'.openi-aitemplate'?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .openi-aitemplate | 45 ++++++++++++++++++++++++++++++--------------- 1 file changed, 30 insertions(+), 15 deletions(-) diff --git a/.openi-aitemplate b/.openi-aitemplate index d4f79fb..efc19b2 100644 --- a/.openi-aitemplate +++ b/.openi-aitemplate @@ -187,18 +187,18 @@ "Name": "NPU训练任务", "Description": "", "JobType": "TRAIN", - "Cluster": "OpenI", + "Cluster": "C2Net", "ComputeSource": "NPU", "HasInternet": 2, "AccCardsNum": 1, "AccCardType": "ASCEND910", "CpuCores": 24, - "MemGiB": 256, + "MemGiB": 96, "GPUMemGiB": 32, "ShareMemGiB": 0, "Image": { - "ImageID": "-2", - "ImageName": "mindspore2.0-cann6", + "ImageID": "0b79fed7504c4e2dacms22cann7train", + "ImageName": "mindspore_2_2_cann7_train", "ImageUrl": "" }, "PretrainModelList": [ @@ -215,12 +215,7 @@ ], "BranchName": "master", "BootFile": "npu_mnist_example/train.py", - "Parameters": [ - { - "Label": "epoch_size", - "Value": "5" - } - ] + "Parameters": [] }, { "Name": "GPU训练任务", @@ -252,7 +247,7 @@ "DatasetName": "MnistDataset_torch.zip" } ], - "BranchName": "liuzx", + "BranchName": "master", "BootFile": "gpu_mnist_example/train.py", "Parameters": [] }, @@ -274,8 +269,18 @@ "ImageName": "", "ImageUrl": "swr.cn-south-1.myhuaweicloud.com/grampus/iluvatar:bi_corex3.1.1_pytorch1.13.1_py3.9_v1.0" }, - "PretrainModelList": [], - "DatasetList": [], + "PretrainModelList": [ + { + "ID": "f1711e48-65d6-436c-b34b-9698061979e7", + "ModelName": "Torch_MNIST_Example_Model" + } + ], + "DatasetList": [ + { + "ID": "8d5178b2-eb56-4c0d-b532-be8c685b1e61", + "DatasetName": "MnistDataset_torch.zip" + } + ], "BranchName": "master", "BootFile": "gpgpu_mnist_example/train.py", "Parameters": [ @@ -303,8 +308,18 @@ "ImageName": "", "ImageUrl": "swr.cn-south-1.myhuaweicloud.com/grampus/enflame:t20_v1.0" }, - "PretrainModelList": [], - "DatasetList": [], + "PretrainModelList": [ + { + "ID": "f1711e48-65d6-436c-b34b-9698061979e7", + "ModelName": "Torch_MNIST_Example_Model" + } + ], + "DatasetList": [ + { + "ID": "8d5178b2-eb56-4c0d-b532-be8c685b1e61", + "DatasetName": "MnistDataset_torch.zip" + } + ], "BranchName": "master", "BootFile": "gcu_mnist_example/train.py", "Parameters": [