[Wait for #2607] [ Layer ] Mixed Precision support for BN Layer#2615

Closed

jijoongmoon wants to merge 17 commits intonnstreamer:mainfrom jijoongmoon:bn_mixed_precision

+2,127-302

Commits on May 7, 2024

Commits on May 8, 2024

[ Test ] Mixed Precision Test Case
jijoongmoon
committed

Commits on May 9, 2024

[ Optimizer ] Update Optimizer / Adam to support Mixed training
jijoongmoon
committed

Commits on May 10, 2024

[ Tensor ] add is_NaN check in Tensor
jijoongmoon
committed

Commits on May 11, 2024

[ Context ] Add loss scale in Context & using mse loss
jijoongmoon
committed

Commits on May 13, 2024

[ Mixed Precision ] Enable Mixed Precision
jijoongmoon
committed

Commits on May 14, 2024

[ Tensor ] Add inifinity check in Tensor
jijoongmoon
committed

Commits on May 17, 2024

[ MSE ] Fix for better MSE loss precision
jijoongmoon
committed

Commits on May 20, 2024

[ TEST ] Add Torch Mixed Precision Model Test
jijoongmoon
committed

Commits on May 22, 2024

[ TEST ] add torch input and output test data for mixed precision
jijoongmoon
committed

Commits on May 24, 2024

[ TEST ] Add more unittest and fixes for mixed precsion
jijoongmoon
committed