@pipcook/plugins-tensorflow-resnet-model-define

Description

This plugin defines a image classification model. This model is resnet. The model is built using python tensorflow.

Required Parameters

| Parameter | Type | Description | |:----------|:-----|:------------|

Optional Parameters

| Parameter | Type | Default Value | Description | |:----------|:-----|:------|:-----| |loss|string|categorical_crossentropy|the loss function to be defined. Look more| |metrics|string[]|[ 'accuracy' ]|metrics used to evaluate model. Look more| |learningRate|number|0.001|base learning rate for Adam| |decay|number|0.05|decay rate for Adam| |freeze|boolean|false|Whether to freeze first several layers| |labelMap|object|null|label map, only specified when recoverPath is provided| |recoverPath|string|null|Recover path of pretrained model|

Example

"modelDefine": {
  "package": "@pipcook/plugins-tensorflow-resnet-model-define",
  "params": {
    "freeze": true
  }
},