from lazy_config.utils.config import LazyCall as L from projects.lazy_config_demo.model import SimpleModule MODEL = L(SimpleModule)( in_features=128, out_features=256, activation="sigmoid" )