Skip to content

Instantly share code, notes, and snippets.

@dpoulopoulos

dpoulopoulos/torch_package_3.py

Created September 29, 2021 15:46
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save dpoulopoulos/d648d5ab4f18c8d906bf5fd4e93d18aa to your computer and use it in GitHub Desktop.
Save dpoulopoulos/d648d5ab4f18c8d906bf5fd4e93d18aa to your computer and use it in GitHub Desktop.
Download ZIP
Raw
torch_package_3.py
from torch import package
path = "/tmp/dcgan.pt"
package_name = "dcgan"
resource_name = "model.pkl"
with package.PackageExporter(path) as exp:
exp.save_pickle(package_name, resource_name, model)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment