xla
30458aaa - Catch exception from AMP mismatch. (#6177)

Commit
2 years ago
Catch exception from AMP mismatch. (#6177) There're a couple of models (yolov3, pytorch_unet, tacotron2) which have AMP mode enabled. If we run the entire benchmarking suite they will crash the main process though. In this PR we catch this, and save the results to the output file.
Parents
Loading