Open dxoigmn opened 1 year ago
It would be really interesting to see if one could use Adversary
in an lightning Callback
. That is, could we fit Adversary
in Callback.on_train_batch_start
. If we can, then we could make this callback a pip-installable package that would "automatically" enable adversarial training just by pip-installing that package. This is enabled via callback entry points.
Right now MART does two things:
It would be nice to be able to take the 2nd part (attack generation loop) and use it in other libraries like https://github.com/openvinotoolkit/anomalib.