classifier/config.json

{
  "architecture": "causilo-v1.0",
  "format_version": 1,
  "config_version": 1,
  "model": {
    "task": "classification",
    "width": 128,
    "expansion": 2,
    "group_size": 3,
    "frequencies": 16,
    "column_latents": 128,
    "column_heads": 4,
    "column_depths": [
      3,
      3
    ],
    "row_heads": 8,
    "row_latents": 4,
    "row_depths": [
      3,
      3
    ],
    "prediction_heads": 4,
    "prediction_depth": 12,
    "outputs": 10
  }
}