pytorch/caffe2/python/layers
Huayu Li cd4c3b48a6 Add LN after specialzied output embeddings and flexible LCE (#35178)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/35178

* add layer norm (LN) after specialized output embeddings
* add flexible lce inside specialized module

Test Plan:
* unit-tests
  * buck test caffe2/caffe2/fb/dper/layer_models/tests/split_1:sparse_nn_test_4 --
  * buck test caffe2/caffe2/fb/dper/layer_models/tests/split_1:sparse_nn_test_6 --

* workflows
  * flexible lce: f177025325

{F232112501}

  *  LN: f177025301

{F232112982}

Differential Revision: D20586281

fbshipit-source-id: 664e77cb4cb5bec6646cafd2e4afb88aff27df03
2020-04-30 15:32:09 -07:00
..
__init__.py
adaptive_weight.py
add_bias.py
arc_cosine_feature_map.py
batch_huber_loss.py
batch_lr_loss.py
batch_mse_loss.py
batch_normalization.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
batch_sigmoid_cross_entropy_loss.py
batch_softmax_loss.py
blob_weighted_sum.py
bpr_loss.py
bucket_weighted.py
build_index.py
concat.py fix spelling mistake: excpected -> expected 2020-01-17 00:11:08 -08:00
constant_weight.py
conv.py
dropout.py
fc.py Add LN after specialzied output embeddings and flexible LCE (#35178) 2020-04-30 15:32:09 -07:00
fc_with_bootstrap.py
fc_without_bias.py
feature_sparse_to_dense.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
functional.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
gather_record.py
homotopy_weight.py
label_smooth.py
last_n_window_collector.py
layer_normalization.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
layers.py Fix type annotations and make MyPy run on torch/ (#36584) 2020-04-22 14:17:08 -07:00
margin_rank_loss.py
merge_id_lists.py
pairwise_similarity.py fix spelling mistake: excpected -> expected 2020-01-17 00:11:08 -08:00
position_weighted.py
random_fourier_features.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
reservoir_sampling.py
sampling_train.py
sampling_trainable_mixin.py
select_record_by_context.py Fix typos, via a Levenshtein-type corrector (#31523) 2020-01-17 16:03:19 -08:00
semi_random_features.py
sparse_dropout_with_replacement.py
sparse_feature_hash.py
sparse_lookup.py
split.py fix spelling mistake: excpected -> expected 2020-01-17 00:11:08 -08:00
tags.py Move gating part of SparseFeatureGating to local 2020-02-16 14:18:27 -08:00
uniform_sampling.py