MNN/tools/train/transformConfig.json

30 lines
632 B
JSON
Raw Permalink Normal View History

{
"Train": true,
"Loss": {
2023-05-18 19:11:50 +08:00
"op": "loss"
},
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":[]
},
"BatchNorm": {
"momentum":0.99
},
2022-08-12 10:30:48 +08:00
"Shape": {
"input": [1, 3, 224, 224]
}
}