Skip to content

MART v0.5.0

Compare
Choose a tag to compare
@mzweilin mzweilin released this 12 Jul 21:40

What's Changed

Improve Adversary using LightningModule and Trainer from Lightning

Improve LitModular and mart.nn

  • Allow modules to skip CallWith by @dxoigmn in #106
  • Allow LitModular.*_step_log to be a dictionary by @dxoigmn in #168
  • Let loss/preds/target output keys be configurable in LitModular by @dxoigmn in #169
  • Remove *_step_end from LitModular by @dxoigmn in #170
  • Make SequentialDict return outputs from all modules as DotDict by @dxoigmn in #171
  • Don't lookup non-str keys in SequentialDict by @dxoigmn in #174
  • Enable overriding of special args in SequentialDict.forward by @dxoigmn in #175
  • Add _train_mode_ and _inference_mode_ special args to SequentialDict by @dxoigmn in #176
  • Add _call_ special arg to SequentialDict by @dxoigmn in #184
  • Allow LitModular's load_state_dict to accept a str by @dxoigmn in #189

Update dependency

Improve performance

  • Speedup adversarial training with a simple perturber by @mzweilin in #89

Move non-essential functionalities to examples

Improve datamodule

  • Make input a dictionary for multi-modal object detection by @mzweilin in #95
  • Quantize datamodule input by @mzweilin in #112

Miscellaneous