Vinh Nguyen

Automatic Mixed Precision training on GPU for Tensorflow has been recently introduced: https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540 This PR adds GPU automatic mixed precision training to `dopamine` via setting a single OS flag: ```...

🐛 [Bug] DETR network conversion failed

1

## Bug Description DETR network obtained from: ``` model = torch.hub.load('facebookresearch/detr:main', 'detr_resnet50', pretrained=True) ``` Scripting doesn't work with PyTorch. So we're following the tracing path. ``` traced_model = torch.jit.trace(model, [torch.randn((128,...

Adding mixed precision training support

8

This PR adds mixed precision training support using APEX. https://github.com/NVIDIA/apex Automatic mixed precision training makes use of both FP32 and FP16 precisions where appropriate. FP16 operations can leverage the Tensor...

As above, I think it would be helpful to have a C++ example application of how to make use of these custom ops.

Adding GPU automatic mixed precision training

2

Automatic Mixed Precision training on GPU for Tensorflow has been recently introduced: https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540 Automatic mixed precision training makes use of both FP32 and FP16 precisions where appropriate. FP16 operations can...

Automatic Mixed Precision training on GPU for TensorFlow has been recently introduced: https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540 Automatic mixed precision training makes use of both FP32 and FP16 precisions where appropriate. FP16 operations can...

Automatic Mixed Precision training on GPU for Tensorflow has been recently introduced: https://medium.com/tensorflow/automatic-mixed-precision-in-tensorflow-for-faster-ai-training-on-nvidia-gpus-6033234b2540 Automatic mixed precision training makes use of both FP32 and FP16 precisions where appropriate. FP16 operations can...

C++ serving example: adding saved model path

6

This PR extends the current C++ image classification example with a saved model path. The two workflows are thus: - Keras saved model ->TFTRT Python API -> frozen graph ->...