core.modules.scaling.fit

core.modules.scaling.fit#

Copyright (c) Meta Platforms, Inc. and affiliates.

This source code is licensed under the MIT license found in the LICENSE file in the root directory of this source tree.

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#
Contents