MCPcopy Create free account
hub / github.com/00why00/JoDiffusion / preprocess_train

Function preprocess_train

train_ae.py:310–314  ·  view source on GitHub ↗
(examples)

Source from the content-addressed store, hash-verified

308 encode = EncodeBitMap(n=math.ceil(math.log(dataset_cls.num_classes, 2)), ignore_label=dataset_cls.ignore_label)
309
310 def preprocess_train(examples):
311 labels = examples[args.label_column]
312 examples["pixel_values"] = [train_transform(label) for label in labels]
313 examples["bit_pixel_values"] = [encode(label)[0] for label in examples["pixel_values"]]
314 return examples
315
316 def preprocess_val(examples):
317 labels = examples[args.label_column]

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected