utils.train
utils.train
Training utils for checkpoints
Functions
| Name | Description |
|---|---|
| determine_last_checkpoint | Determine the checkpoint to resume from based on configuration. |
determine_last_checkpoint
utils.train.determine_last_checkpoint(cfg, update=True)Determine the checkpoint to resume from based on configuration.
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| cfg | DictDefault | Dictionary mapping axolotl config keys to values. |
required |
| update | bool | Whether to update the config with the determined checkpoint | True |
Returns
| Name | Type | Description |
|---|---|---|
| str | None | Path to the checkpoint to resume from, or None if not resuming. |