google-research / mint

Multi-modal Content Creation Model Training Infrastructure including the FACT model (AI Choreographer) implementation.
Apache License 2.0
497 stars 85 forks source link

Why the seq_name list repeats 10 times during preprocessing.py? #55

Open by2101 opened 2 years ago

by2101 commented 2 years ago

Hi, authors, thank you for your nice work! I am learning your FID calculation code. I found that in Line 170 in preprocessing.py, the seq_name repeats 10 times, so that the evaluator run 10 times on testval set. I am not understand this design. https://github.com/google-research/mint/blob/b8f8bdfbbe3fbfa67831a2ef9bcf71a6a9e74552/tools/preprocessing.py#L170

And can I reproduce the FID scores in your paper with the provided checkpoints?

Thank you! Best