Wrap optimizer in mart.modules.Modular in mart.optim.OptimizerFactory

IntelLabs / MART

Modular Adversarial Robustness Toolkit

BSD 3-Clause "New" or "Revised" License

17 stars 0 forks source link

Wrap optimizer in mart.modules.Modular in mart.optim.OptimizerFactory #151

Closed dxoigmn closed 1 year ago

dxoigmn commented 1 year ago

          Shall we add this to `mart/models/modular.py` so that we can simplify the config in `mart/configs/optimizer`? That will make it easier to add other off-the-shelf optimizers from PyTorch.

_Originally posted by @mzweilin in https://github.com/IntelLabs/MART/pull/147#discussion_r1213437268_