topic = "pytorch"
lesson = 8

from n import *
home, models_path = get_project_dir("FashionMNIST")
print_(home)

import torch
import torchvision.models as models

models_path

# pretrained model
model = models.vgg16(weights=models.VGG16_Weights.DEFAULT)

model_path = f"{models_path}/vgg16.pth"

torch.save(model.state_dict(), model_path)
print_("model saved: " + model_path)

# No default weights loaded
model = models.vgg16()

# Weights loaded from saved weights
model.load_state_dict(torch.load(model_path))

model.eval();