wchen61 comments

Repositories
Issues
Comments

Results 1 comments of


                                            wchen61

[BERT/PyTorch] Unable to reproduce bert benchmark under A100

Thank you for this information. @sharathts I have modify this torch._C._jit_set_autocast_mode(True), and can be run successfully. When changed batch size to 512, get CUDA OOM. Traceback (most recent call last):...