Как исправить эту RuntimeError? - PullRequest
Новая
       11

Как исправить эту RuntimeError?

0 голосов
/ 27 июня 2019

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

Файл train.py, включающий контрольную точку сохранения, работает полностью. Этот второй файл Forex.py не запускается 

def load_checkpoint(filepath):
    print(filepath)
    **checkpoint = torch.load(filepath)**
    print(checkpoint.keys())
    if checkpoint['arch'] == p_ague.arch:
        model = models.vgg19(pretrained= True)
    if checkpoint['arch'] == p_ague.arch:
        model = models.alexnet(pretrained= True)
    for param in model.parameters():
        param.requires_grad = False
    else:
        print('Model Not recognised')

    model.class_to_idx = checkpoint['class_to_idx']

    model.classifier = nn.Sequential(nn.Linear(p_ague.input_units, p_ague.hidden_units),
                                 nn.Dropout(p_ague.dropout),
                                 nn.ReLU(),
                                 nn.Linear(p_ague.hidden_units,  p_ague.output_units),
                                 nn.LogSoftmax(dim=1))
    model.load_state_dict(checkpoint['state_dict'])
    return model

**model = load_checkpoint(p_ague.checkpoint)**

RuntimeError: хранилище имеет неправильный размер: 0 получено 102

...