When initializing your task, simply make sure to turn off the auto-connect for pytorch, using Task.init(..., auto_connect_frameworks={'pytorch': False})

thanks

however, this will also turn off metrics 😞

oh okay

it would be useful to have some control over this if possible 🙂

RobustRat47 it seems I was mistaken - for anything other than tensorboard and matplotlib, it only controls models binding

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),