core.modules.scaling.fit

Attributes

parser

Functions

_prefilled_input(→ str)
_train_batch(→ None)
compute_scaling_factors(→ None)

Module Contents

core.modules.scaling.fit._prefilled_input(prompt: str, prefill: str = '') → str

core.modules.scaling.fit._train_batch(trainer: fairchem.core.trainers.base_trainer.BaseTrainer, batch) → None

core.modules.scaling.fit.compute_scaling_factors(config, num_batches: int = 16) → None

core.modules.scaling.fit.parser