Coverage for src / autoencodix / configs / vanillix_config.py: 100%

1from .default_config import DefaultConfig 

2from pydantic import Field 

3 

4 

5class VanillixConfig(DefaultConfig): 

6 """ 

7 A specialized configuration inheriting from DefaultConfig. 

8 """ 

9 

10 beta: float = Field( 

11 default=0.1, # Overridden default (was 1.0) 

12 ge=0, 

13 description="Beta weighting factor for VAE loss", 

14 ) 

15 epoch: int = Field( 

16 default=30, ge=0, description="How many epochs should the model train for." 

17 ) 

18 # TODO find sensible defaults for Vanillix