lunar-navigation / classify /layers.json
sarapohland's picture
uploading classification model
9f7df74 verified
raw
history blame contribute delete
966 Bytes
[
{
"name": "conv",
"n_in": 3,
"n_out": 32,
"kernel": 5,
"stride": 2,
"pad": 2,
"activation": "relu"
},
{
"name": "conv",
"n_in": 32,
"n_out": 64,
"kernel": 3,
"stride": 2,
"pad": 1,
"activation": "relu"
},
{
"name": "conv",
"n_in": 64,
"n_out": 64,
"kernel": 3,
"stride": 2,
"pad": 1,
"activation": "relu"
},
{
"name": "flatten"
},
{
"name": "fc",
"n_in": 76800,
"n_out": 512,
"activation": "relu"
},
{
"name": "fc",
"n_in": 512,
"n_out": 512,
"activation": "relu"
},
{
"name": "fc",
"n_in": 512,
"n_out": 256,
"activation": "relu"
},
{
"name": "fc",
"n_in": 256,
"n_out": 256,
"activation": "relu"
},
{
"name": "fc",
"n_in": 256,
"n_out": 8,
"activation": "softmax"
}
]