python - numpy.load() 未加载

Question

我想保存我训练过的参数，以便稍后加载并使用它们。不幸的是，这对我不起作用，这是我的代码：

import os.path

def save(parameters):
    #parameters have weights and biases
    L = len(parameters)//2
    W = {}
    b = {}
    for l in range(1,L):
        W["W" + str(l)] = parameters["W" + str(l)]
        b["b" + str(l)] = parameters["b" + str(l)]
    print(W)
    num = 1
    while os.path.isfile('trainedData/weights{}.npy'.format(num)):
        num += 1 
    else:
        np.save('trainedData/weights{}.npy'.format(num), W)

    num = 1
    while os.path.isfile('trainedData/biases{}.npy'.format(num)):
        num += 1
    else:
        np.save('trainedData/biases{}.npy'.format(num), b)

save(parameters)
#PRINT(W) IS CORRECTLY PRINTING WEIGHTS

def load(num):
    W = np.load('trainedData/weights{}.npy'.format(num), allow_pickle=True)
    b = np.load('trainedData/biases{}.npy'.format(num), allow_pickle=True)
    print(W)

load(1)

将打印这个：

{}

我究竟做错了什么？谢谢你的帮助！

标签： pythonnumpyneural-network