9 lines
204 B
Python
9 lines
204 B
Python
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"
|
|
) |