search

YyzHarry / imbalanced-regression

[ICML 2021, Long Talk] Delving into Deep Imbalanced Regression
http://dir.csail.mit.edu
MIT License
826 stars 130 forks source link

how to update statistics in FDS #29

Closed huangbingyang2020 closed 2 years ago

huangbingyang2020 commented 2 years ago

when update statistics in FDS, do we need to calculate the mean and variance of all sample in each bin after every epoch finishes? do we need to update the sum of z and z^2 after each mini-batch?

YyzHarry commented 2 years ago

See https://github.com/YyzHarry/imbalanced-regression/blob/main/agedb-dir/train.py#L269-L281