shape-prediction-model / config.json
sdtemple's picture
Push model using huggingface_hub.
d0f24d5 verified
{
"dropout": 0.2,
"height": 224,
"hidden_dim": 16,
"kernel_size": 3,
"num_classes": 4,
"num_cnn_channels": 16,
"num_cnn_layers": 3,
"num_input_channels": 3,
"num_layers": 1,
"padding": 1,
"pooling": 2,
"stride": 1,
"width": 224
}