Skip to content
This repository has been archived by the owner on Oct 23, 2023. It is now read-only.

pytorch.proto and respective changes #61

Merged
merged 5 commits into from
May 26, 2020

Conversation

igorvalko
Copy link
Contributor

@igorvalko igorvalko commented May 8, 2020

TL;DR

proto files and generated ones for pytorch Flyte plugin

Type

  • Bug Fix
  • Feature
  • Plugin

Are all requirements met?

  • Code completed
  • Smoke tested
  • Unit tests added
  • Code documentation added
  • Any pending items have an associated Issue

Complete description

That's the part of pytorch Flyte plugin implementation.
Related PRs:
flyteorg/flytekit#112
flyteorg/flyteplugins#85

Tracking Issue

N/A

Follow-up issue

N/A


option go_package = "github.com/lyft/flyteidl/gen/pb-go/flyteidl/plugins";

message PyTorchOperatorTask {
Copy link
Contributor

Choose a reason for hiding this comment

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

Can you add some code comments here...

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Sure, done.

Copy link
Contributor

Choose a reason for hiding this comment

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

I think we should call the plugin type, DistributedPytorchTraining?

Copy link
Contributor

Choose a reason for hiding this comment

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

Also the comment then should be, this plugin enables distributed training for pytorch using "link to the Kubeflow operator" - What do you think @igorvalko ?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

done

@wild-endeavor wild-endeavor merged commit 15de3cd into flyteorg:master May 26, 2020
eapolinario pushed a commit that referenced this pull request Sep 8, 2023
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants