IntelLabs / MART

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

Add FiftyOne callbacks #213

Open lumurillo opened 11 months ago

lumurillo commented 11 months ago

What does this PR do?

Add two different FiftyOne callbacks:

The predictions and evaluation metrics can be automatically visualized into the FiftyOne app: image

This PR is dependant of PR #209

Type of change

Please check all relevant options.

Testing

Please describe the tests that you ran to verify your changes. Consider listing any relevant details of your test configuration.

Before submitting

Known issues

When using the GPU device, the trainer.predict throws the following error:

RuntimeError: Input type (torch.cuda.FloatTensor) and weight type (torch.FloatTensor) should be the same

The input data is on GPU, but the model's weights is on CPU.