本文整理匯總了Python中models.resnext.get_fine_tuning_parameters方法的典型用法代碼示例。如果您正苦於以下問題：Python resnext.get_fine_tuning_parameters方法的具體用法？Python resnext.get_fine_tuning_parameters怎麽用？Python resnext.get_fine_tuning_parameters使用的例子？那麽, 這裏精選的方法代碼示例或許可以為您提供幫助。您也可以進一步了解該方法所在類models.resnext的用法示例。
在下文中一共展示了resnext.get_fine_tuning_parameters方法的1個代碼示例，這些例子默認根據受歡迎程度排序。您可以為喜歡或者感覺有用的代碼點讚，您的評價將有助於係統推薦出更棒的Python代碼示例。
示例1: generate_model
# 需要導入模塊: from models import resnext [as 別名]
# 或者: from models.resnext import get_fine_tuning_parameters [as 別名]
def generate_model( opt):
assert opt.model in ['resnext']
assert opt.model_depth in [101]
from models.resnext import get_fine_tuning_parameters
model = resnext.resnet101(
num_classes=opt.n_classes,
shortcut_type=opt.resnet_shortcut,
cardinality=opt.resnext_cardinality,
sample_size=opt.sample_size,
sample_duration=opt.sample_duration,
input_channels=opt.input_channels,
output_layers=opt.output_layers)
model = model.cuda()
model = nn.DataParallel(model)
if opt.pretrain_path:
print('loading pretrained model {}'.format(opt.pretrain_path))
pretrain = torch.load(opt.pretrain_path)
assert opt.arch == pretrain['arch']
model.load_state_dict(pretrain['state_dict'])
model.module.fc = nn.Linear(model.module.fc.in_features, opt.n_finetune_classes)
model.module.fc = model.module.fc.cuda()
parameters = get_fine_tuning_parameters(model, opt.ft_begin_index)
return model, parameters
return model, model.parameters()