mnist_example / config.json
Wauplin's picture
Wauplin HF staff
Save model using Keras.
2355b67 verified
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
28,
28,
1
],
"dtype": "float32",
"sparse": false,
"name": "input_layer"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 32,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "valid",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
28,
28,
1
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "max_pooling2d",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
26,
26,
32
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"filters": 64,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "valid",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
13,
13,
32
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "max_pooling2d_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
11,
11,
64
]
}
},
{
"module": "keras.layers",
"class_name": "Flatten",
"config": {
"name": "flatten",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"data_format": "channels_last"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
5,
5,
64
]
}
},
{
"module": "keras.layers",
"class_name": "Dropout",
"config": {
"name": "dropout",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"rate": 0.5,
"seed": null,
"noise_shape": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
1600
]
}
},
{
"module": "keras.layers",
"class_name": "Dense",
"config": {
"name": "dense",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "FloatDTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null,
"shared_object_id": 140283585819024
},
"units": 10,
"activation": "softmax",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
1600
]
}
}
],
"build_input_shape": [
null,
28,
28,
1
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
28,
28,
1
]
}
}