IntelLabs / MART

Modular Adversarial Robustness Toolkit
BSD 3-Clause "New" or "Revised" License
16 stars 0 forks source link

Assume `LitModular` by default in configuration #190

Open dxoigmn opened 1 year ago

dxoigmn commented 1 year ago

Right now our configuration is rather verbose when it comes to models:

model:
  _target_: mart.models.LitModular
  _convert_: all
  output_preds_key: preds
  output_target_key: target
  modules:
    ...
  training_sequence:
    ...
  ...

We could clean this up by assuming LitModular by default and moving all keys under model up 1 level like:

output_preds_key: preds
output_target_key: target
modules:
  ...
training_sequence:
  ...
...