Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/imgcook/pipcook-plugin-tensorflow-image-classification-model-train

The Pipcook model train plugin for tensorflow-based image classification tasks.
https://github.com/imgcook/pipcook-plugin-tensorflow-image-classification-model-train

Last synced: 3 days ago
JSON representation

The Pipcook model train plugin for tensorflow-based image classification tasks.

Awesome Lists containing this project

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
```json
"modelTrain": {
"package": "@pipcook/plugins-image-classification-tensorflow-model-train",
"params": {
"epochs": 15
}
}
```