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)

собирается напечатать только это:

{}

Что я делаю не так? Спасибо за любую помощь!