@pipcook/plugins-image-classification-tensorflow-model-train
Description
This plugin will be used to train image classification models built on top of tensorflow.
Required Parameters
Parameter |
Type |
Description |
Optional Parameters
Parameter |
Type |
Default Value |
Description |
epochs |
number |
10 |
how many epochs to train |
batchSize |
number |
16 |
data's batch size to be trained for every iteration |
Example
"modelTrain": {
"package": "@pipcook/plugins-image-classification-tensorflow-model-train",
"params": {
"epochs": 15
}
}