pytorch
1189f559 - Creating new layer FCWithBootstrap used in bootstrapping uncertainty approach (#29152)

Commit
5 years ago
Creating new layer FCWithBootstrap used in bootstrapping uncertainty approach (#29152) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/29152 Bootstrapping uncertainty approach: bootstrap the last layer before the last fully-connected layer. FCWithBootstrap is a new layer to handle the logic for the bootstrapping process. Goal: - return a struct with the bootstrapped indices and bootstrapped predictions from this layer - separate the functionality in the train_net and eval_net - save the bootstrapped FC in this object so that the eval_net can use them during prediction time Reviewed By: wx1988 Differential Revision: D17822429 fbshipit-source-id: 15dec501503d581aeb69cb9ae9e8c3a3fbc7e7b5
Author
Kevin Chen
Parents
Loading