toolkit / verify_tags.py
k4d3's picture
Update num_classes in JTP2Processor to 18166 for alignment with state_dict
c28cd7a
raw
history blame
156 Bytes
import json
with open("/home/kade/source/repos/JTP2/tags.json", "r", encoding="utf-8") as f:
tags = json.load(f)
print(f"Number of tags: {len(tags)}")