config.json

{
  "architecture": "RN50",
  "device": 0,
  "embed_dim": 1024,
  "pool_out": "mean",
  "pooling": "avgpool",
  "queue_data": null,
  "queue_size": 256,
  "trainable_layers": [
    "visual"
  ],
  "transformers_version": "4.45.2"
}