diff --git a/routers/repo/aisafety.go b/routers/repo/aisafety.go
index f845c3918..eb3019e5a 100644
--- a/routers/repo/aisafety.go
+++ b/routers/repo/aisafety.go
@@ -351,6 +351,7 @@ func DelAiSafetyTask(ctx *context.Context) {
func AiSafetyCreateForGetGPU(ctx *context.Context) {
ctx.Data["PageIsCloudBrain"] = true
ctx.Data["IsCreate"] = true
+ ctx.Data["datasetType"] = models.TypeCloudBrainOne
ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName
ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID
ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.CombatDataSetName
@@ -362,6 +363,7 @@ func AiSafetyCreateForGetGPU(ctx *context.Context) {
func AiSafetyCreateForGetNPU(ctx *context.Context) {
ctx.Data["PageIsCloudBrain"] = true
ctx.Data["IsCreate"] = true
+ ctx.Data["datasetType"] = models.TypeCloudBrainTwo
ctx.Data["BaseDataSetName"] = setting.ModelSafetyTest.BaseDataSetName
ctx.Data["BaseDataSetUUID"] = setting.ModelSafetyTest.BaseDataSetUUID
ctx.Data["CombatDataSetName"] = setting.ModelSafetyTest.CombatDataSetName
diff --git a/routers/repo/cloudbrain.go b/routers/repo/cloudbrain.go
index 1f6732174..b1020c082 100755
--- a/routers/repo/cloudbrain.go
+++ b/routers/repo/cloudbrain.go
@@ -51,8 +51,8 @@ const (
tplCloudBrainBenchmarkNew base.TplName = "repo/cloudbrain/benchmark/new"
tplCloudBrainBenchmarkShow base.TplName = "repo/cloudbrain/benchmark/show"
- tplCloudBrainModelSafetyNewGpu base.TplName = "repo/cloudbrain/modelsafety/newgpu"
- tplCloudBrainModelSafetyNewNpu base.TplName = "repo/cloudbrain/modelsafety/newnpu"
+ tplCloudBrainModelSafetyNewGpu base.TplName = "repo/modelsafety/newgpu"
+ tplCloudBrainModelSafetyNewNpu base.TplName = "repo/modelsafety/newnpu"
tplCloudBrainImageSubmit base.TplName = "repo/cloudbrain/image/submit"
tplCloudBrainImageEdit base.TplName = "repo/cloudbrain/image/edit"
diff --git a/templates/repo/cloudbrain/benchmark/new.tmpl b/templates/repo/cloudbrain/benchmark/new.tmpl
index 863edd2d0..13665c036 100755
--- a/templates/repo/cloudbrain/benchmark/new.tmpl
+++ b/templates/repo/cloudbrain/benchmark/new.tmpl
@@ -51,7 +51,7 @@
{{.i18n.Tr "repo.cloudbrain.benchmark.model"}}
模型安全评测
+ href="{{.RepoLink}}/modelsafety/create_gpu">模型安全评测
@@ -298,7 +298,7 @@
{{.i18n.Tr "repo.cloudbrain.benchmark.model"}}
模型安全评测
+ href="{{.RepoLink}}/modelsafety/create_gpu">模型安全评测
diff --git a/templates/repo/modelsafety/indicators_descr.tmpl b/templates/repo/modelsafety/indicators_descr.tmpl
new file mode 100644
index 000000000..3b6f7ff48
--- /dev/null
+++ b/templates/repo/modelsafety/indicators_descr.tmpl
@@ -0,0 +1,62 @@
+
+
+
+
+
diff --git a/templates/repo/modelsafety/new.tmpl b/templates/repo/modelsafety/new.tmpl
index b2d180215..1b6cccf46 100644
--- a/templates/repo/modelsafety/new.tmpl
+++ b/templates/repo/modelsafety/new.tmpl
@@ -49,19 +49,42 @@
@@ -237,9 +221,6 @@
if(createFlag) return false
createFlag = true
}
- let repolink = {{.RepoLink }}
- let url_href = window.location.pathname.split('create')[0]
- $(".ui.button").attr('href', url_href)
$('.menu .item').tab();
var isValidate = false;
@@ -285,7 +266,7 @@
})
}
- validate()
+ // validate()
$('.ui.create_train_job.green.button').click(function (e) {
validate()
})
@@ -303,14 +284,39 @@
})();
;(function() {
- $('.indicators_descr').popup({
- inline: true,
- hoverable: true,
- position: 'top center',
- delay: {
- show: 300,
- hide: 800
+ var baseDataSetName = {{.BaseDataSetName}} || '';
+ var baseDataSetUUID = {{.BaseDataSetUUID}} || '';
+ var combatDataSetName = {{.CombatDataSetName}} || '';
+ var combatDataSetUUID = {{.CombatDataSetUUID}} || '';
+ var baseDataSetSel = $('#baseDataSet-sel');
+ var combatDataSetSel = $('#combatDataSet-sel');
+
+ function initDataSetSelect(names, keys, selectEl) {
+ selectEl.empty();
+ var nameList = names.split(';');
+ var keyList = keys.split(';')
+ for (var i = 0, iLen = nameList.length; i < iLen; i++) {
+ selectEl.append(``);
}
- });
+ selectEl.on('change', function() {
+ var self = $(this);
+ var selectIndex = self.find('option:selected').attr('index');
+ var type = self.attr('id');
+ if (type === 'baseDataSet-sel') {
+ var val = combatDataSetSel.find('option[index="'+selectIndex+'"]').attr('value');
+ combatDataSetSel.dropdown('set selected', val);
+ } else {
+ var val = baseDataSetSel.find('option[index="'+selectIndex+'"]').attr('value');
+ baseDataSetSel.dropdown('set selected', val);
+ }
+ }).trigger('change');
+ }
+ initDataSetSelect(baseDataSetName, baseDataSetUUID, baseDataSetSel);
+ initDataSetSelect(combatDataSetName, combatDataSetUUID, combatDataSetSel);
+
+ $('#job_indicators').on('change', function() {
+ var value = $(this).val();
+ $('input[name="evaluationIndex"]').val(value.join(';'));
+ }).trigger('change');
})();
diff --git a/templates/repo/modelsafety/newgpu.tmpl b/templates/repo/modelsafety/newgpu.tmpl
new file mode 100644
index 000000000..d41c8f816
--- /dev/null
+++ b/templates/repo/modelsafety/newgpu.tmpl
@@ -0,0 +1 @@
+{{template "repo/modelsafety/new".}}
diff --git a/templates/repo/modelsafety/newnpu.tmpl b/templates/repo/modelsafety/newnpu.tmpl
new file mode 100644
index 000000000..d41c8f816
--- /dev/null
+++ b/templates/repo/modelsafety/newnpu.tmpl
@@ -0,0 +1 @@
+{{template "repo/modelsafety/new".}}
diff --git a/web_src/js/components/dataset/selectDataset.vue b/web_src/js/components/dataset/selectDataset.vue
index 77d54033f..057298d13 100755
--- a/web_src/js/components/dataset/selectDataset.vue
+++ b/web_src/js/components/dataset/selectDataset.vue
@@ -1011,7 +1011,8 @@ export default {
if (
location.href.indexOf("benchmark") !== -1 ||
location.href.indexOf("train-job") !== -1 ||
- location.href.indexOf("inference") !== -1
+ location.href.indexOf("inference") !== -1 ||
+ location.href.indexOf("modelsafety") !== -1
) {
this.benchmarkNew = true;
}
diff --git a/web_src/js/components/images/selectImages.vue b/web_src/js/components/images/selectImages.vue
index add4553d0..506574655 100755
--- a/web_src/js/components/images/selectImages.vue
+++ b/web_src/js/components/images/selectImages.vue
@@ -505,7 +505,8 @@ export default {
if (
location.href.indexOf("train-job") !== -1 ||
location.href.indexOf("inference-job") !== -1 ||
- location.href.indexOf("benchmark") !== -1
+ location.href.indexOf("benchmark") !== -1 ||
+ location.href.indexOf("modelsafety") !== -1
) {
this.benchmarkNew = true;
}