OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
54b33503ec
pytorch/caffe2/python/layers
History
Huan Gui d3fcd0d798 add dropout during eval (#17549) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/17549

Currently Dropout is only enabled in training, we enable the option of having dropout in Eval.

This is to follow [1]. This functionality would be used for uncertainty estimation in exploration project.

[1] Gal, Yarin, and Zoubin Ghahramani. "Dropout as a bayesian approximation: Representing model uncertainty in deep learning." international conference on machine learning. 2016.

Reviewed By: Wakeupbuddy

Differential Revision: D14216216

fbshipit-source-id: 87c8c9cc522a82df467b685805f0775c86923d8b
2019-02-28 23:21:29 -08:00
..
__init__.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
adaptive_weight.py fbshipit-source-id: ba600fcd2b5cefc7621357bdeb05e24cea02e5af 2018-06-27 04:50:56 -07:00
add_bias.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
arc_cosine_feature_map.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_distill_lr_loss.py Co-disitillation with different archs and/or feature set (#9793) 2018-07-25 10:10:27 -07:00
batch_lr_loss.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
batch_mse_loss.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_normalization.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_sigmoid_cross_entropy_loss.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
batch_softmax_loss.py [GanH] Label Smooth: Add Layer and Integrate to SparseNN 2018-03-27 18:10:39 -07:00
blob_weighted_sum.py Update from facebook (#8384) 2018-06-13 13:10:45 -07:00
bucket_weighted.py Implement bucket-based attention pooling for IdScoreList features (#13004) 2018-10-25 18:04:08 -07:00
build_index.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
concat.py refactor preproc, support dense in TumHistory layer 2018-09-05 16:10:13 -07:00
constant_weight.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
conv.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
dropout.py add dropout during eval (#17549) 2019-02-28 23:21:29 -08:00
fc_without_bias.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
fc.py fc layer accept axis argument (#13822) 2018-11-11 13:44:57 -08:00
feature_sparse_to_dense.py Revert D13551909: [fbcode] logdevice for generic feature type 2019-01-25 00:33:06 -08:00
functional.py Update from facebook (#7855) 2018-05-29 11:38:02 -07:00
gather_record.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
homotopy_weight.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
label_smooth.py [GanH] Label Smooth: Add Layer and Integrate to SparseNN 2018-03-27 18:10:39 -07:00
last_n_window_collector.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
layer_normalization.py Allow use substitute ops for LayerNorm (#12177) 2018-10-11 17:36:10 -07:00
layers.py [Caffe2][fbcode=>GH sync] Update from facebook 4323b18ce13c (#7116) 2018-05-01 20:49:00 -07:00
margin_rank_loss.py Allow calculating average margin rank loss 2018-03-20 13:34:22 -07:00
merge_id_lists.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
pairwise_similarity.py move matrix formation for dot products to precompute/request-only (#10531) 2018-08-15 11:02:10 -07:00
position_weighted.py Remove unused code base for distributed training (#10282) 2018-08-16 20:10:17 -07:00
random_fourier_features.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
reservoir_sampling.py [Caffe2] Changes done inside Facebook (#6378) 2018-04-10 21:11:43 -07:00
sampling_train.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sampling_trainable_mixin.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
select_record_by_context.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
semi_random_features.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
sparse_feature_hash.py Update from facebook (#7451) 2018-05-10 23:14:27 -07:00
sparse_lookup.py Resubmit: Set the correct engine name for position weighted pooling when fp16 is used for training 2018-11-27 14:51:56 -08:00
split.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00
tags.py Remove unused code base for distributed training (#10282) 2018-08-16 20:10:17 -07:00
uniform_sampling.py Remove Apache headers from source. 2018-03-27 13:10:18 -07:00