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

[StyleGen] Classifier on generator: BART bug #3716

Merged
merged 1 commit into from
Jun 15, 2021
Merged

Conversation

emilydinan
Copy link
Contributor

@emilydinan emilydinan commented Jun 14, 2021

Patch description
Fix classifier on generator code to work for BART. The code overrides the TransformerDecoder, but does not include the forward through the norm_embeddings if variant == 'bart'

This bug was caught when we do DDP -- DDP expects a grad for the norm_embeddings after forwarding but none is found

Copy link
Contributor

@EricMichaelSmith EricMichaelSmith left a comment

Choose a reason for hiding this comment

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

looks good!

@emilydinan emilydinan merged commit e93a00f into master Jun 15, 2021
@emilydinan emilydinan deleted the classtgabart branch June 15, 2021 14:02
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants