Skip to content
This repository was archived by the owner on Jul 1, 2024. It is now read-only.

Mixed precision training #338

Closed

Conversation

mannatsingh
Copy link
Contributor

Summary: This adds a set_amp_opt_level() function to ClassificationTask, which can be used to train with apex.amp.

Differential Revision: D18331200

Summary: This adds a `set_amp_opt_level()` function to `ClassificationTask`, which can be used to train with `apex.amp`.

Differential Revision: D18331200

fbshipit-source-id: f09a4320f8933734d1e834ddf155aa3d3cd70ccf
@facebook-github-bot facebook-github-bot added CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported labels Jan 13, 2020
@facebook-github-bot
Copy link
Contributor

This pull request was exported from Phabricator. Differential Revision: D18331200

facebook-github-bot pushed a commit that referenced this pull request Jan 14, 2020
Summary:
Pull Request resolved: #338

This adds a `set_amp_opt_level()` function to `ClassificationTask`, which can be used to train with `apex.amp`.

Reviewed By: vreis

Differential Revision: D18331200

fbshipit-source-id: af9018bc2119b7ae0f1a2042d6f6d8d6c4fdb715
miguelvr pushed a commit to miguelvr/ClassyVision that referenced this pull request Jan 15, 2020
Summary:
Pull Request resolved: facebookresearch#338

This adds a `set_amp_opt_level()` function to `ClassificationTask`, which can be used to train with `apex.amp`.

Reviewed By: vreis

Differential Revision: D18331200

fbshipit-source-id: af9018bc2119b7ae0f1a2042d6f6d8d6c4fdb715
@vreis vreis mentioned this pull request Feb 11, 2020
@mannatsingh mannatsingh deleted the export-D18331200 branch June 29, 2020 21:30
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants