diff --git a/.nojekyll b/.nojekyll index dce54c08f..3313d6c22 100644 --- a/.nojekyll +++ b/.nojekyll @@ -1 +1 @@ -10d9ca4d \ No newline at end of file +c1062b54 \ No newline at end of file diff --git a/docs/api/core.trainers.base.html b/docs/api/core.trainers.base.html index dfe865ff3..a7ba2bc98 100644 --- a/docs/api/core.trainers.base.html +++ b/docs/api/core.trainers.base.html @@ -511,10 +511,6 @@ pre > code.sourceCode > span > a:first-child::before { text-decoration: underlin
push_to_hub method in order to force-add the tags when pushing thecore.trainers.base.AxolotlTrainer.push_to_hub(*args, **kwargs)Overwrite the push_to_hub method in order to force-add the tags when pushing the
model on the Hub. Please refer to ~transformers.Trainer.push_to_hub for more details.
core.trainers.base.AxolotlTrainer.training_step(
- model,
- inputs,
- num_items_in_batch=None,
-)Perform a training step on a batch of inputs. Overrides the
-transformers.trainer.Trainer method to handle sequence parallelism if
-enabled.
| Name | -Type | -Description | -Default | -
|---|---|---|---|
| model | -nn.Module | -Model to perform training step for. | -required | -
| inputs | -dict[str, torch.Tensor | Any] | -Dictionary mapping. | -required | -