config.json

{
  "model_type": "bucket-detection",
  "input_channels": 8,
  "input_size": [
    8,
    96,
    96
  ],
  "embedding_dim": 512,
  "output_shape": [
    5,
    5
  ],
  "backbone": "resnet18",
  "random_init": true
}