Finegrain/refiners
1
0
Fork 0
mirror of https://github.com/finegrain-ai/refiners.git synced 2024-11-24 23:28:45 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix training_101 import

Browse source
This commit is contained in:
limiteinductive 2024-04-18 18:21:55 +00:00 committed by Benjamin Trom
parent 446796da57
commit 07985694ed
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
docs/guides/training_101/index.md
View file

@ -219,7 +219,7 @@ We will now define the configuration for the autoencoder. It holds the configura
Example: Example:
```python ```python
from refiners.training_utils import BaseConfig, TrainingConfig, OptimizerConfig, LRSchedulerConfig, Optimizers, LRSchedulers, Epoch from refiners.training_utils import BaseConfig, TrainingConfig, OptimizerConfig, LRSchedulerConfig, Optimizers, LRSchedulerType, Epoch
class AutoencoderConfig(BaseConfig): class AutoencoderConfig(BaseConfig):
# Since we are using a synthetic dataset, we will use a arbitrary fixed epoch size. # Since we are using a synthetic dataset, we will use a arbitrary fixed epoch size.
@ -238,7 +238,7 @@ optimizer = OptimizerConfig(
) )
lr_scheduler = LRSchedulerConfig( lr_scheduler = LRSchedulerConfig(
type=LRSchedulers.ConstantLR type=LRSchedulerType.ConstantLR
) )
config = AutoencoderConfig( config = AutoencoderConfig(