2019-06-17 20:10:35 +08:00
|
|
|
{
|
|
|
|
|
"Train": true,
|
|
|
|
|
"Loss": {
|
2023-05-18 19:11:50 +08:00
|
|
|
"op": "loss"
|
2019-06-17 20:10:35 +08:00
|
|
|
},
|
2023-05-18 19:11:50 +08:00
|
|
|
"Optimizer": {
|
2022-08-12 10:30:48 +08:00
|
|
|
"OnlyUpdateOps":[],
|
|
|
|
|
"NoUpdateOps":[],
|
2022-11-08 17:05:14 +08:00
|
|
|
"StopBackPropOps":[],
|
2023-05-18 19:11:50 +08:00
|
|
|
"type": "SGD",
|
|
|
|
|
"ParameterOptConfig":[
|
|
|
|
|
{
|
|
|
|
|
"WeightNames":["example_Weight1", "example_Weight2"],
|
|
|
|
|
"LrName":"LearningRate2"
|
|
|
|
|
},
|
|
|
|
|
{
|
|
|
|
|
"WeightNames":["example_Weight3"],
|
|
|
|
|
"LrName":"LearningRate3"
|
|
|
|
|
}
|
|
|
|
|
],
|
|
|
|
|
"FixAsConstOps":[]
|
2019-06-17 20:10:35 +08:00
|
|
|
},
|
2022-12-04 15:17:36 +08:00
|
|
|
"BatchNorm": {
|
|
|
|
|
"momentum":0.99
|
|
|
|
|
},
|
2022-08-12 10:30:48 +08:00
|
|
|
"Shape": {
|
|
|
|
|
"input": [1, 3, 224, 224]
|
2019-06-17 20:10:35 +08:00
|
|
|
}
|
|
|
|
|
}
|