TimelyPenguin76 SuccessfulKoala55 Thank you. I found an error, the wrong port was specified. Now I don't have a display of saved models in the 'models'tab. I save it like this:
` model_jit = torch.jit.trace(model, torch.ones(1, 3, 480, 480).to('cuda'))

model_jit.save(os.path.join(checkpoint_path, f'{epoch_num}_{round(acc_full, 4)}.pt')) `