fiddle_config_quickstart_model_optimizer_build.py

1import fiddle as fdl
2
3class Optimizer:
4  def __init__(self, learning_rate: float):
5    self.learning_rate = learning_rate
6
7class Model:
8  def __init__(self, optimizer: Optimizer, layers: int):
9    self.optimizer = optimizer
10    self.layers = layers
11
12def base_config():
13  config = fdl.Config(Model)
14  config.optimizer = fdl.Config(Optimizer, learning_rate=0.01)
15  config.layers = 12
16  return config
17
18# Create the configuration
19config = base_config()
20
21# Materialize (build) the actual objects
22model = fdl.build(config)
23
24print(f"Model layers: {model.layers}")
25print(f"Optimizer LR: {model.optimizer.learning_rate}")