StefanHex comments on Try training token-level probes

[StefanHex]

Thanks! Fixed