Skip to content

Allow amp training#15

Open
vovaf709 wants to merge 2 commits intoRobinBruegger:masterfrom
vovaf709:amp
Open

Allow amp training#15
vovaf709 wants to merge 2 commits intoRobinBruegger:masterfrom
vovaf709:amp

Conversation

@vovaf709
Copy link
Copy Markdown

Recomputation of intermediate activations fails in the case of mixed precision training. Simple torch.cuda.amp.autocast fixes it.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant