Hi @Tanya_Boone

torch.save(model,‘model1.pth’)
AttributeError: Can’t pickle local object ‘_initialize…patch_forward…new_fwd’

seems like your model can not be saved with torch.save.
Maybe you need to replace some lambda function in there, if there are some…
Can you share the model definition to take a look ?