Runners

class compressors.runners.HFRunner(*args, **kwargs)

Runner for supervised task for Huggingface transformers.

handle_batch(batch)

Inner method to handle specified data batch. Used to make a train/valid/infer stage during Experiment run.

Parameters

batch (Mapping[str, Any]) – dictionary with data batches from DataLoader.