This repository has been archived by the owner on Jul 1, 2024. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 278
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Add support to attach heads to DenseNets (#383)
Summary: Pull Request resolved: #383 Added the following attachable blocks to the `DenseNet` implementation- Contains the following attachable blocks - - `block{block_idx}-{idx}` - `transition-{idx}` - `trunk_output` The `trunk_output` block is the final output of the `DenseNet`. This is where a `fully_connected` head will normally be attached. Also added an additional arg (`zero_init_bias`) to `FullyConnectedHead` since `DenseNets` init the fc bias to 0. Reviewed By: vreis Differential Revision: D19780372 fbshipit-source-id: f52bafc7a6d03f4f6678d2726918b23769ddeb3f
- Loading branch information
1 parent
8575416
commit 183f518
Showing
4 changed files
with
97 additions
and
52 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters