Collator# class stable_pretraining.data.Collator(G_from=None)[source]# Bases: object Custom collate function that optionally builds an affinity (or “graph”) matrix based on a specified field.