Skip to content

Support multiple outputs in pytorch parser#1151

Merged
vloncar merged 6 commits intofastmachinelearning:mainfrom
JanFSchulte:multiout_pytorch
Feb 7, 2025
Merged

Support multiple outputs in pytorch parser#1151
vloncar merged 6 commits intofastmachinelearning:mainfrom
JanFSchulte:multiout_pytorch

Conversation

@JanFSchulte
Copy link
Contributor

Fixes the issue reported in #1147 that the pytorch parser was not able to handle multiple outputs and would just use the last node as the output layer.

Type of change

  • Bug fix (non-breaking change that fixes an issue)

-## Tests

Tested the example model provided in #1147, which fails to produce the correct output before this PR and is successful after this fix.

Checklist

  • I have read the guidelines for contributing.
  • I have commented my code, particularly in hard-to-understand areas.
  • I have made corresponding changes to the documentation.
  • My changes generate no new warnings.
  • I have installed and run pre-commit on the files I edited or added.
  • I have added tests that prove my fix is effective or that my feature works.

@JanFSchulte
Copy link
Contributor Author

pre-commit.ci autofix

@JanFSchulte JanFSchulte added bug please test Trigger testing by creating local PR branch labels Dec 13, 2024
@JanFSchulte JanFSchulte added please test Trigger testing by creating local PR branch and removed please test Trigger testing by creating local PR branch labels Dec 13, 2024
@JanFSchulte JanFSchulte added please test Trigger testing by creating local PR branch and removed please test Trigger testing by creating local PR branch labels Dec 14, 2024
@JanFSchulte JanFSchulte requested a review from vloncar December 16, 2024 13:47
@JanFSchulte JanFSchulte added please test Trigger testing by creating local PR branch and removed please test Trigger testing by creating local PR branch labels Feb 7, 2025
@vloncar vloncar merged commit 3b7e595 into fastmachinelearning:main Feb 7, 2025
5 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

bug please test Trigger testing by creating local PR branch

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants