mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +00:00
fix training_101 import
This commit is contained in:
parent
446796da57
commit
07985694ed
|
@ -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(
|
||||||
|
|
Loading…
Reference in a new issue