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.

Sign up for GitHub

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

Jump to bottom

could i use pytorch model code on tfx #59

Closed
DanlanChen opened this issue Apr 25, 2019 · 1 comment
Closed

could i use pytorch model code on tfx #59

DanlanChen opened this issue Apr 25, 2019 · 1 comment
Assignees
@krazyhaas
Labels
stat:awaiting response type:feature

Comments

@DanlanChen
Copy link

could i use pytorch model code on tfx
@krazyhaas
Copy link

Not very easily, but you're welcome to try and we'll help out where we can. One of the challenges will be that TFX the intermediate data as TFRecords. I'm not a pytorch expert but I think you'd need to first convert the TFRecords into torch.Tensors before the pytorch dataloader could consume the output of the upstream components.

ruoyu90 pushed a commit to ruoyu90/tfx that referenced this issue Aug 28, 2019
@BryanCutler @ewilderj
Add Bryan to SIG IO release team (tensorflow#59)
70dc6de
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@krazyhaas krazyhaas

Labels
stat:awaiting response type:feature
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
3 participants
@DanlanChen @krazyhaas @gowthamkpr