Skip to content

fix model output_schema dims for BC/Regression task case #646

Merged
rnyak merged 5 commits intomainfrom
fix_model_output_schema
Mar 15, 2023
Merged

fix model output_schema dims for BC/Regression task case #646
rnyak merged 5 commits intomainfrom
fix_model_output_schema

Conversation

@rnyak
Copy link
Copy Markdown
Contributor

@rnyak rnyak commented Mar 14, 2023

This PR fixes the dims info in the output_schema based on the task and summary_type. The current dims for the BC task is not accurately set.

Better to merge this after this PR #641

@rnyak rnyak added this to the Merlin 23.03 milestone Mar 14, 2023
@rnyak rnyak requested a review from sararb March 14, 2023 01:11
@rnyak rnyak added the chore Maintenance for the repository label Mar 14, 2023
@github-actions
Copy link
Copy Markdown

@rnyak rnyak merged commit 279e09d into main Mar 15, 2023
@rnyak rnyak deleted the fix_model_output_schema branch March 15, 2023 18:39
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area/api area/pytorch chore Maintenance for the repository

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants