code: def save(self, destination, **kwargs):
          
          with open(destination, "wb") as f:
              torch.save(self, f, **kwargs)
docs: Serialize and save a model.
  
          Example:
              end_model = EndModel(...)
              end_model.train_model(...)
              end_model.save("my_end_model.pkl")