@pipcook/plugins-image-classification-tensorflow-model-train
v1.1.4
Published
### Description
Downloads
44
Keywords
Readme
@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
}
}