{ "embedding_size": 1024, "hidden_sizes": [ 512, 256 ], "init_constant": 0.01, "input_shape": [ 3, 448, 448 ] }