import pickle

def save_model(model, file_name):
    with open(file_name, 'wb') as f:
        pickle.dump(model.get_weights(), f)

def load_model(mode, file_name):
    with open(file_name, 'rb') as f:
        attn_model.set_weights(pickle.load(f))