NSAPH-Projects / topological-equivariant-networks

E(n)-Equivariant Topological Neural Networks
MIT License
19 stars 0 forks source link

Hydra ML config manager instead of Argparse #18

Closed mauriciogtec closed 2 months ago

mauriciogtec commented 7 months ago

Use Hydra instead of Argparse, since it is easier to manage configs of different models, datasets, etc.

For example, iinstead of instantiating models with lengthy code, such as:

if args.model == "....":
   model = ....

We can use a single call

model = instantiate(cfg.model)
mauriciogtec commented 3 months ago

currently used by the etnn_spatial

audiracmichelle commented 2 months ago

Closed in 77bf3ba