|
|
|
@@ -1544,14 +1544,14 @@ func NewContext() { |
|
|
|
|
|
|
|
func getModelSafetyConfig() { |
|
|
|
sec := Cfg.Section("model_safety_test") |
|
|
|
ModelSafetyTest.GPUBaseDataSetName = sec.Key("GPUBaseDataSetName").MustString("ImageNet1000_100基础数据集;CIFAR10_1000基础数据集") |
|
|
|
ModelSafetyTest.GPUBaseDataSetUUID = sec.Key("GPUBaseDataSetUUID").MustString("20e5e8c8-58fd-4c15-95ff-3811da3ec200;18fc091a-cc00-4862-b04a-dfa798a14909") |
|
|
|
ModelSafetyTest.GPUCombatDataSetName = sec.Key("GPUCombatDataSetName").MustString("ImageNet1000_100_FGSM;CIFAR10_1000_FGSM.zip") |
|
|
|
ModelSafetyTest.GPUCombatDataSetUUID = sec.Key("GPUCombatDataSetUUID").MustString("3e76f133-d9c6-4819-976d-4487d41f4d6d;787b561d-57bd-4ecf-b833-ef26492cdd87") |
|
|
|
ModelSafetyTest.NPUBaseDataSetName = sec.Key("NPUBaseDataSetName").MustString("ImageNet1000_100基础数据集;CIFAR10_1000基础数据集") |
|
|
|
ModelSafetyTest.NPUBaseDataSetUUID = sec.Key("NPUBaseDataSetUUID").MustString("20e5e8c8-58fd-4c15-95ff-3811da3ec200;18fc091a-cc00-4862-b04a-dfa798a14909") |
|
|
|
ModelSafetyTest.NPUCombatDataSetName = sec.Key("NPUCombatDataSetName").MustString("ImageNet1000_100_FGSM;CIFAR10_1000_FGSM.zip") |
|
|
|
ModelSafetyTest.NPUCombatDataSetUUID = sec.Key("NPUCombatDataSetUUID").MustString("3e76f133-d9c6-4819-976d-4487d41f4d6d;787b561d-57bd-4ecf-b833-ef26492cdd87") |
|
|
|
ModelSafetyTest.GPUBaseDataSetName = sec.Key("GPUBaseDataSetName").MustString("") |
|
|
|
ModelSafetyTest.GPUBaseDataSetUUID = sec.Key("GPUBaseDataSetUUID").MustString("") |
|
|
|
ModelSafetyTest.GPUCombatDataSetName = sec.Key("GPUCombatDataSetName").MustString("") |
|
|
|
ModelSafetyTest.GPUCombatDataSetUUID = sec.Key("GPUCombatDataSetUUID").MustString("") |
|
|
|
ModelSafetyTest.NPUBaseDataSetName = sec.Key("NPUBaseDataSetName").MustString("") |
|
|
|
ModelSafetyTest.NPUBaseDataSetUUID = sec.Key("NPUBaseDataSetUUID").MustString("") |
|
|
|
ModelSafetyTest.NPUCombatDataSetName = sec.Key("NPUCombatDataSetName").MustString("") |
|
|
|
ModelSafetyTest.NPUCombatDataSetUUID = sec.Key("NPUCombatDataSetUUID").MustString("") |
|
|
|
} |
|
|
|
|
|
|
|
func getModelConvertConfig() { |
|
|
|
|