config.json

{"dim": 512, "dim_output": 1, "n_layers": 12, "n_heads": 4, "task": "REGRESSION"}