Function save_data

tools/utils/save_load.py:97–101

(fetch_batch_var, save_path, prefix='result')

def save_data(fetch_batch_var, save_path, prefix='result'):
    logger.info("start save batch data")
    save_data = np.array(fetch_batch_var, dtype=object)
    model_prefix = os.path.join(save_path, prefix)
    np.save(model_prefix, save_data)