conditional-flow-matching
conditional-flow-matching copied to clipboard
Add a generate samples script
What does this PR do?
Add a script to generate multiple images for the CIFAR-10 examples
Fixes #73
Before submitting
- [x] Did you make sure title is self-explanatory and the description concisely explains the PR?
- [x] Did you make sure your PR does only one thing, instead of bundling different changes together?
- [ ] Did you list all the breaking changes introduced by this pull request?
- [x] Did you test your PR locally with
pytestcommand? - [ ] Did you run pre-commit hooks with
pre-commit run -acommand?
Codecov Report
All modified and coverable lines are covered by tests :white_check_mark:
Comparison is base (
3c0cfdd) 35.90% compared to head (6562e19) 35.90%.
:exclamation: Current head 6562e19 differs from pull request most recent head c41f83d. Consider uploading reports for the commit c41f83d to get more accurate results
Additional details and impacted files
@@ Coverage Diff @@
## main #84 +/- ##
=======================================
Coverage 35.90% 35.90%
=======================================
Files 67 67
Lines 7419 7419
=======================================
Hits 2664 2664
Misses 4755 4755
:umbrella: View full report in Codecov by Sentry.
:loudspeaker: Have feedback on the report? Share it here.
Since the generation might be time-consuming, we could also try to add a multi-GPU support. I took a look at this script https://github.com/NVlabs/edm/blob/main/generate.py but it is unclear how to do it