chrisjcundy's picture
Add probe weights: 7 models x {linear, mlp, efc, axial}
b8131eb verified
Raw History Blame Contribute Delete
128 Bytes
{
"class_name": "LinearProbe",
"init_args": {
"d_model": 4096,
"nhead": 1,
"normalize_input": "unit_norm"
}
}