fail to load pre-model

[mk123qwe]

    model = torch.load(‘mbv3_small.pth.tar‘, map_location='cpu') 
    print('Loading base network...')
    weight = model["state_dict"]
    base_net = torch.nn.DataParallel(base_net)
    base_net.load_state_dict(weights)

The program has been running, no response

[Buster-maker]

you can try it

model = torch.load('./mbv3_small.pth', map_location = 'cpu')
    print('Loading base network...')
    weight = model["state_dict"]
    base_net = torch.nn.DataParallel(net)
    base_net.load_state_dict(weight)
    y = net(x)
    end = time.time()
    print('Running time: %s Seconds' % (end - start))
    print(y.size())