FX-based feature extraction is a new TorchVision utility that allows access to intermediate transformations of an input during the forward pass of a PyTorch Module.
β PyTorch (@PyTorch) October 29, 2021
Read more below on what makes TorchVision's utility more versatile than existing methods.https://t.co/RRsowlXPwi