wrapping feature calculation in the inference_mode() instead of (face…

…bookresearch#5)

no_grad()

Marginal speed-up on the probing task

Co-authored-by: Evgeny Kharitonov <[email protected]>