Сохранить и загрузить numpy ndarray в виде текстового файла (веса keras)

Я хочу использовать свою модель keras на разных компьютерах с разными версиями Python.

Я не хочу использовать Pickle и numpy.savez, потому что это вызывает проблемы в разных средах.

Я начинаю с этого, и это работает нормально

import json
import numpy as np
from tensorflow import keras

def save_mod(model, name = "my_model"):
    with open(name + '.json', 'w') as fp:
        json.dump(model.to_json(), fp)

    # save weights
    model_weights = model.get_weights()
    return model_weights

def load_mod(model_weights_, name = "my_model"):
    # load config
    with open(name + ".json", "r") as read_file:
        json_string = json.load(read_file)
    model_ = keras.models.model_from_json(json_string, custom_objects = {})

    # load weights
    model_.set_weights(model_weights_)
    return model_

model = keras.models.load_model("segmentation.h5")
weights = save_mod(model)
loaded_model = load_mod(weights)

Затем я пытаюсь сохранить и загрузить веса

np.savetxt('weights.txt', weights, fmt='%s')
loaded_weights = np.fromfile('weights.txt')

print(len(weights)) 
print(len(loaded_weights))

>> 112
>> 31013

У меня есть 112 и 31013, метод не работает

with open('test.txt', 'wb') as f:
    np.savetxt(f, np.column_stack(weights), fmt='%1.10f')

он говорит >> все входные массивы должны иметь одинаковое количество измерений, но массив с индексом 0 имеет 4 измерения, а массив с индексом 1 имеет 2 измерения.

наконец я сделал это

class EncodeNumpy(json.JSONEncoder):
    def default(self, obj):
        if isinstance(obj, np.ndarray):
            return obj.tolist()
        return json.JSONEncoder.default(self, obj)


we = np.array(weights)
print(a.shape)
json_dump = json.dumps({'we': we}, cls=EncodeNumpy)

json_load = json.loads(json_dump)
a_restored = np.asarray(json_load["we"])
print(a_restored.shape)

model.set_weights(a_restored)

это дало мне массивы (112) и (112)

но model.set_weights(a_restored) возвращает

AttributeError: 'list' object has no attribute 'shape'