From 07985694eda9bbdaaa6a1dd4cd0b673b208cdf34 Mon Sep 17 00:00:00 2001 From: limiteinductive Date: Thu, 18 Apr 2024 18:21:55 +0000 Subject: [PATCH] fix training_101 import --- docs/guides/training_101/index.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/guides/training_101/index.md b/docs/guides/training_101/index.md index 8d1be89..0b62806 100644 --- a/docs/guides/training_101/index.md +++ b/docs/guides/training_101/index.md @@ -219,7 +219,7 @@ We will now define the configuration for the autoencoder. It holds the configura Example: ```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): # Since we are using a synthetic dataset, we will use a arbitrary fixed epoch size. @@ -238,7 +238,7 @@ optimizer = OptimizerConfig( ) lr_scheduler = LRSchedulerConfig( - type=LRSchedulers.ConstantLR + type=LRSchedulerType.ConstantLR ) config = AutoencoderConfig(