pytorch-sbs/settings/CIAR10/CNN/network.json

97 lines
2 KiB
JSON
Raw Normal View History

2023-01-15 14:11:38 +01:00
{
"network_structure": {
"number_of_output_neurons": 10,
"layer_type": [
"CONV2D",
"RELU",
"MAX POOLING",
"CONV2D",
"RELU",
"MAX POOLING",
"CONV2D",
"RELU",
"CONV2D"
],
"strides": [
[
1,
1
], // "CONV2D"
[], // "RELU",
[
2,
2
], // "MAX POOLING",
[
1,
1
], // "CONV2D"
[], // "RELU",
[
2,
2
], // "MAX POOLING",
[
1,
1
], // "CONV2D"
[], // "RELU",
[
1,
1
] // "CONV2D"
],
"forward_neuron_numbers": [
[
3,
32
], // "CONV2D",
[], // "RELU",
[], // "MAX POOLING",
[
32,
64
], // "CONV2D",
[], // "RELU",
[], // "MAX POOLING",
[
64,
96
], // "CONV2D",
[], // "RELU",
[
96,
10
] // "CONV2D",
],
"forward_kernel_size": [
[
5,
5
], // "CONV2D",
[], // "RELU",
[
2,
2
], // "MAX POOLING",
[
5,
5
], // "CONV2D",
[], // "RELU",
[
2,
2
], // "MAX POOLING",
[
4,
4
], // "CONV2D",
[], // "RELU",
[
1,
1
] // "CONV2D",
]
}
}