MNN/tools/train/transformConfig.json

30 lines
632 B
JSON

{
"Train": true,
"Loss": {
"op": "loss"
},
"Optimizer": {
"OnlyUpdateOps":[],
"NoUpdateOps":[],
"StopBackPropOps":[],
"type": "SGD",
"ParameterOptConfig":[
{
"WeightNames":["example_Weight1", "example_Weight2"],
"LrName":"LearningRate2"
},
{
"WeightNames":["example_Weight3"],
"LrName":"LearningRate3"
}
],
"FixAsConstOps":[]
},
"BatchNorm": {
"momentum":0.99
},
"Shape": {
"input": [1, 3, 224, 224]
}
}