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 

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

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