Set requires grad for FID metric

pytorch / torcheval

A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.

https://pytorch.org/torcheval

Other

211 stars 46 forks source link

Set requires grad for FID metric #167

Closed ananthsub closed 1 year ago

ananthsub commented 1 year ago

Summary: As title, set requires grad for model parameters in the FID metric constructor. This takes care of one more thing callers who provide a custom module have to do

Differential Revision: D47351250

facebook-github-bot commented 1 year ago

This pull request was exported from Phabricator. Differential Revision: D47351250