however, this will also turn off metrics
For the sake of future readers, let me clarify on this one, turning it off auto_connect_frameworks={'pytorch': False}
only effects the auto logging of torch.save/load
(side note: the reason is pytorch does not have built in metric reporting, i.e. it is usually done manually and these days most probably with tensorboard, for example lightning / ignite will use tensorboard as default metric reporting),
RobustRat47 it seems I was mistaken - for anything other than tensorboard and matplotlib, it only controls models binding
it would be useful to have some control over this if possible 🙂
however, this will also turn off metrics 😞
When initializing your task, simply make sure to turn off the auto-connect for pytorch, using Task.init(..., auto_connect_frameworks={'pytorch': False})