How can I use FlashTorch to see what the EfficientNet learn?

[ChenXiao61]

Describe the bug A clear and concise description of what the bug is.

To Reproduce Steps to reproduce the behavior:

1. Go to '...'
2. Click on '....'
3. Scroll down to '....'
4. See error

Expected behavior A clear and concise description of what you expected to happen.

Screenshots If applicable, add screenshots to help explain your problem.

Environment (please complete the following information):

• OS: [e.g. iOS]
• Python version: [e.g. 3.6]
• FlashTorch version: [e.g. 0.1.0]

Additional context Add any other context about the problem here.

[MisaOgura]

@ChenXiao61 The pre-trained EfficientNet models are available from https://github.com/lukemelas/EfficientNet-PyTorch#loading-pretrained-models

Once you install the package and follow the instruction to load a model, you should be able to use it with the Backprop class to create saliency maps like any other models loaded from torchvision.models.