12 lines
266 B
Python
12 lines
266 B
Python
|
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'], {}))
|