Skip to content

patch: minor fix in q value networks#120

Closed
neuralsorcerer wants to merge 1 commit intofacebookresearch:mainfrom
neuralsorcerer:patch-2
Closed

patch: minor fix in q value networks#120
neuralsorcerer wants to merge 1 commit intofacebookresearch:mainfrom
neuralsorcerer:patch-2

Conversation

@neuralsorcerer
Copy link
Contributor

Change:

  • The CNNQValueMultiHeadNetwork.get_q_values method now checks len(action_batch.shape) when deciding whether to squeeze the returned tensor instead of using len(action_batch).

Why?

  • The previous logic misinterpreted a 3‑D action tensor: len(action_batch) returned the batch size rather than the number of dimensions, causing incorrect squeezing behavior and potentially wrong output shapes.

@meta-cla meta-cla bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Jul 18, 2025
@facebook-github-bot
Copy link
Contributor

@Yonathae has imported this pull request. If you are a Meta employee, you can view this in D78899714.

@facebook-github-bot
Copy link
Contributor

@rodrigodesalvobraz merged this pull request in 4ddce21.

@neuralsorcerer neuralsorcerer deleted the patch-2 branch July 25, 2025 19:00
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. Merged

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants