The definition of 'load_lua' cannot be found

[Dendrobium123]

[OaDsis]

Thanks for your attention. You can load the pre-trained model of vgg16 more easily, like this:

import torch import torchvision.models as models

def init_vgg16():

Initialize the VGG16 model with pretrained weights

vgg = models.vgg16(pretrained=True)
return vgg

Use the function to get the model

vgg_model = init_vgg16()