How to chnage the Feature extractor?

[ghost]

Hi there,

I am intersted to understand how can we chnage the feature extractor of FID to another networks like ResNet18?

Here's the code

under the paramete it is written feature (Union[int, Module]) where an nn.Module for using a custom feature extractor. Expects that its forward method returns an (N,d) matrix where N is the batch size and d is the feature size.

Does anyone know how to replace InceptionV3 to Resnet or any other model?

Thanks