search

toshas / torch-fidelity

High-fidelity performance metrics for generative models in PyTorch
Other
1.01k stars 66 forks source link

Add a new tf feature extractor #23

Closed dougsouza closed 3 years ago

dougsouza commented 3 years ago

Hello,

I have tf inception model finetuned specific for my dataset. I would like to use this model to compute IS and FID using torch fidelity. How do I convert this inception model to torch and preserve behavior as you did?

Can you share how you did it?

Thanks,

Douglas

toshas commented 3 years ago

If it is the same model, use this script: https://github.com/toshas/torch-fidelity/blob/master/utils/util_convert_inception_weights.py

If the model is different, you will also need to create a compatible torch.nn.Module