facebookresearch / mae

PyTorch implementation of MAE https//arxiv.org/abs/2111.06377

Is possible to enable FP16 or TF32 in pretraining?

Wongboo opened this issue a year ago · comments

Wongboo commented a year ago

Is possible to enable FP16 or TF32 in pretraining?

Mang Ning commented 4 months ago

I have the same question, anyone know?