custom-torch-model / config.json
1
{"in_dim": 2, "out_dim": 6}