def __init__(self, args):
super(CustomResnet, self).__init__()
layers = get_layers(args.block_layout)
self._model = ResNet(layers, args)
model_name = args.pretrained_imagenet_model_name
if args.pretrained_on_imagenet:
load_pretrained_weights(self._model,
load_pretrained_model(model_name))
def __init__(self, args):
super(Default_Resnet50, self).__init__()
block_layout = [[('Bottleneck', 3)], [('Bottleneck', 4)],
[('Bottleneck', 6)], [('Bottleneck', 3)]]
layers = get_layers(block_layout)
self._model = ResNet(layers, args)
if args.pretrained_on_imagenet:
load_pretrained_weights(self._model,
load_pretrained_model('resnet50'))
def __init__(self, args):
super(Default_8StageResnet36, self).__init__()
block_layout = [[('BasicBlock', 2)], [('BasicBlock', 2)],
[('BasicBlock', 2)], [('BasicBlock', 2)],
[('BasicBlock', 2)], [('BasicBlock', 2)],
[('BasicBlock', 2)], [('BasicBlock', 2)]]
layers = get_layers(block_layout)
self._model = ResNet(layers, args)
if args.pretrained_on_imagenet:
load_pretrained_weights(self._model,
load_pretrained_model('resnet18'))
