DL-Art-School/codes/models/classifiers/torch_models.py

12 lines
266 B
Python
Raw Normal View History

from torchvision.models import vgg16
from trainer.networks import register_model
from utils.util import opt_get
@register_model
def register_torch_vgg16(opt_net, opt):
""" return a ResNet 18 object
"""
return vgg16(**opt_get(opt_net, ['kwargs'], {}))