| { | |
| "type": "sequential", | |
| "layers": "Sequential(\n (0): Linear(in_features=1024, out_features=256, bias=True)\n (1): ReLU()\n (2): Dropout(p=0.1, inplace=False)\n (3): Linear(in_features=256, out_features=1, bias=True)\n)", | |
| "num_labels": 1, | |
| "hidden_size": 1024 | |
| } |