config.json

{
  "depth": 6,
  "device": "cuda:0",
  "dimensions_c": 64,
  "dimensions_f": 64,
  "dimensions_s": 64,
  "input_dim": 1024,
  "nheads": 16,
  "num_clusters": 100,
  "num_outputs": 20820
}