KaihuaTang / Long-Tailed-Recognition.pytorch

[NeurIPS 2020] This project provides a strong single-stage baseline for Long-Tailed Classification, Detection, and Instance Segmentation (LVIS). It is also a PyTorch implementation of the NeurIPS 2020 paper 'Long-Tailed Classification by Keeping the Good and Removing the Bad Momentum Causal Effect'.
GNU General Public License v3.0
560 stars 68 forks source link

Moving average of embedding #14

Closed rahulvigneswaran closed 3 years ago

rahulvigneswaran commented 3 years ago

Why is there a moving average of the embedding out of the feature before it goes into the fully connected network? What is it used for? https://github.com/KaihuaTang/Long-Tailed-Recognition.pytorch/blob/540cae6da251f49b1b021630e219dde7c5867ce2/classification/run_networks.py#L215

KaihuaTang commented 3 years ago

Please refer to our paper: https://arxiv.org/abs/2009.12991