Add configurable number of gen steps

[jacobbieker]

Pull Request

Description

This can help with the large amount of GPU memory used.

Fixes #22

How Has This Been Tested?

Unit tests

• [ ] Yes

If your changes affect data processing, have you plotted any changes? i.e. have you done a quick sanity check?

• [ ] Yes

Checklist:

• [ ] My code follows OCF's coding style guidelines
• [ ] I have performed a self-review of my own code
• [ ] I have made corresponding changes to the documentation
• [ ] I have added tests that prove my fix is effective or that my feature works
• [ ] I have checked my code and corrected any misspellings

[codecov-commenter]

Codecov Report

Merging #40 (586d542) into main (8eb9b21) will increase coverage by 0.06%. The diff coverage is 33.33%.

@@            Coverage Diff             @@
##             main      #40      +/-   ##
==========================================
+ Coverage   72.10%   72.17%   +0.06%     
==========================================
  Files          12       12              
  Lines         839      841       +2     
==========================================
+ Hits          605      607       +2     
  Misses        234      234              

Impacted Files	Coverage Δ
dgmr/dgmr.py	40.14% <33.33%> (+0.42%)	:arrow_up:
dgmr/discriminators.py	100.00% <0.00%> (ø)

:mega: We’re building smart automated test selection to slash your CI/CD build times. Learn more