convert_to_perfclient.py

import torch
from torch_cmspepr.dataset import TauDataset
import json
from tqdm import tqdm

dataset = TauDataset("/mnt/d/mldata/hgcalml/npzs_all")

out_list = []

for i, data in tqdm(enumerate(dataset[:1000])):
  out_list.append({
    "features__0": { "content": data.x.flatten().tolist(), "shape": list(data.x.size()) },
    "batch__0": { "content": torch.zeros((data.x.size()[0])).tolist(), "shape": [data.x.size()[0]] },
  })


with open("perfdata.json", "w") as fout:
  json.dump({"data": out_list}, fout)