config.json

{
  "model_type": "collect-model",
  "input_channels": 6,
  "input_size": [
    6,
    96,
    96
  ],
  "embedding_dim": 512,
  "output_shape": [
    20,
    3
  ],
  "backbone": "resnet18",
  "random_init": true
}