Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add MetNetPV training script #45

Draft
wants to merge 10 commits into
base: main
Choose a base branch
from
Draft

Conversation

jacobbieker
Copy link
Member

Pull Request

Description

Adds training script for the new MetNetPV model version

Fixes #

How Has This Been Tested?

Please describe the tests that you ran to verify your changes. Provide instructions so we can reproduce.
Please also list any relevant details for your test configuration

  • 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

@jacobbieker jacobbieker self-assigned this Nov 16, 2022
@jacobbieker
Copy link
Member Author

Going to attempt a few more tests -> 1. Try MetNet that outputs all future timesteps at once 'MetNetSingleShot', 2. Run with just sun positions to see if it can learn the general trends of night time, 3. Run with smaller inputs (64x64) to fit all 96 timesteps in at once. For all of these, also try running with multiple GPUs on donatello, the new torchdata might have fixed some issues

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants