Как я могу извлечь функции с нуля из модели, которая не прошла предварительную подготовку - PullRequest
Новая
       20

Как я могу извлечь функции с нуля из модели, которая не прошла предварительную подготовку

0 голосов
/ 11 июля 2019

Я пытаюсь извлечь элементы из модели, которая не прошла предварительную подготовку, то есть, словами, я хотел бы использовать свой собственный набор данных.

Я пытался следовать предварительно подготовленному примеру, но не могу заставить его работать. 

scratch_model = initialize_model(model_name, num_classes, feature_extract=False, use_pretrained=False)
scratch_model = scratch_model.to(device)
scratch_optimizer = optim.SGD(scratch_model.parameters(), lr=0.001, momentum=0.9)
scratch_criterion = nn.CrossEntropyLoss()
scratch_hist = train_model(scratch_model, dataloaders_dict, scratch_criterion, scratch_optimizer, num_epochs=num_epochs, is_inception=(model_name=="resnet"))

Инициализация модели для этого прогона 

scratch_model, input_size = initialize_model(model_name, num_classes, feature_extract, use_pretrained=False)
...