train_model.py 文件源码

def save(model, optimizer, save_name, args):
    serializers.save_npz(save_name + "model", copy.deepcopy(model).to_cpu())
    serializers.save_npz(save_name + "optimizer", optimizer)
    print('save', save_name)