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

PyTorch Predictor #1051

Merged
merged 17 commits into from
Sep 28, 2020
Merged

PyTorch Predictor #1051

merged 17 commits into from
Sep 28, 2020

Conversation

PascalIversen
Copy link
Contributor

@PascalIversen PascalIversen commented Sep 22, 2020

Description of changes:

  1. added a PyTorchPredictor class
  2. added gluonts.torch.support.util.copy_parameters function

based on PR #1052

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

Copy link
Contributor

@lostella lostella left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Thanks for this! I have only a few comments inline for now.

For the notebook: before we can merge it you should make sure to clear all cells, so that the state of the notebook is not checked into the repo (this makes the file much more lightweight in general)

src/gluonts/model/predictor.py Outdated Show resolved Hide resolved
@lostella
Copy link
Contributor

lostella commented Sep 23, 2020

@PascalIversen one more comment: maybe it's better to split this into multiple PRs

  1. moving the existing code,
  2. introducing the new predictor type (this PR),
  3. adding the example the notebook.

I'm checking how quick this can be done, but it seems that if you reset the branch to master and only add files outside of src/gluonts/torch you should quickly get item 1.

Let me know if you have time for this, otherwise I can take care of 1, and rebase the current PR on top of that.

src/gluonts/model/forecast_generator.py Outdated Show resolved Hide resolved
src/gluonts/model/forecast_generator.py Outdated Show resolved Hide resolved
src/gluonts/model/predictor.py Outdated Show resolved Hide resolved
PascalIversen and others added 3 commits September 28, 2020 15:44
Co-authored-by: Jasper <jasper.b.schulz@googlemail.com>
Co-authored-by: Jasper <jasper.b.schulz@googlemail.com>
Co-authored-by: Jasper <jasper.b.schulz@googlemail.com>
@lostella lostella dismissed their stale review September 28, 2020 14:39

changed my mind

@lostella lostella merged commit bcb2ef2 into awslabs:master Sep 28, 2020
kashif pushed a commit to kashif/gluon-ts that referenced this pull request Oct 10, 2020
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.

3 participants