mirror of
https://github.com/clearlinux/graphene.git
synced 2026-09-05 21:31:36 +00:00
Dayeol Lee
e651c1d262
- This example will be compared with another example for loading encrypted models and inputs using PFS. Thus, load the model from a file. - Updated Makefile and README.md
9 lines
232 B
Python
9 lines
232 B
Python
from torchvision import models
|
|
import torch
|
|
|
|
output_filename = "alexnet-pretrained.pt"
|
|
alexnet = models.alexnet(pretrained=True)
|
|
torch.save(alexnet, output_filename)
|
|
|
|
print("Pre-trained model was saved in \"%s\"" % output_filename)
|