Neural Network Compression Framework for enhanced OpenVINO™ inference

nlp sparsity compression deep-learning tensorflow transformers pytorch classification pruning object-detection quantization semantic-segmentation bert onnx openvino mixed-precision-training quantization-aware-training llm genai

• Updated Nov 9, 2026
• Python

Implementation of our Pattern Recognition paper "DMT: Dynamic Mutual Training for Semi-Supervised Learning"

pytorch semi-supervised-learning tensorboard cifar semantic-segmentation pascal-voc cityscapes mixed-precision-training

• Updated Jun 16, 2022
• Python

A Natural Language Inference (NLI) model based on Transformers (BERT and ALBERT)

natural-language-processing albert natural-language-inference bert nli mixed-precision-training nli-model

• Updated Feb 4, 2024
• Python

This repository containts the pytorch scripts to train mixed-precision networks for microcontroller deployment, based on the memory contraints of the target device.

pytorch integer-arithmetic qnn low-power-mcu quantized-neural-networks edge-ai mixed-precision-training

• Updated May 9, 2024
• Python

transformer imagenet neural-machine-translation cifar10 pytorch-implementation multi30k mixed-precision-training distributed-data-parallel vision-transformer

• Updated Dec 31, 2022
• Python

A deep learning project for sport image classification using a custom VGG19-based architecture with integrated Grad-CAM heatmap visualization for model interpretability.

computer-vision deep-learning cuda grad-cam pytorch gpu-acceleration image-classification transfer-learning vgg19 data-augmentation explainable-ai sports-analytics mixed-precision-training heatmap-visualization sports-classification pytorch-grad-cam

• Updated Jun 28, 2026
• Jupyter Notebook

A pytorch helper library for Mixed Precision Training, Initialization, Metrics and More Utilities to simplify training of deep learning models

python machine-learning metrics pytorch nvidia deeplearning initializer mixed-precision-training

• Updated Jun 8, 2019
• Python

YOLOv12 Underwater Object Detection is an open-source suite for underwater object detection, built on YOLOv12. It offers an end-to-end pipeline with GPU-accelerated training, customizable data augmentations, real-time inference via Gradio, and support for model export (ONNX & PyTorch).

computer-vision deep-learning image-processing pytorch gpu-acceleration object-detection gradio evaluation-metrics data-augmentation attention-mechanisms onnx model-export training-pipeline mixed-precision-training underwater-object-detection underwater-imaging real-time-inference yolov12 environmental-adaptation brackish-water

• Updated Jun 17, 2025
• Python

Using Deep Learning To Identify And Classify Building Damage

computer-vision deep-learning pytorch convolutional-neural-networks object-detection unet semantic-segmentation pytorch-cnn unet-image-segmentation mixed-precision-training vision-transformer

• Updated Aug 13, 2024
• Python

Let's train CIFAR 10 Pytorch with Half-Precision!

training cnn pytorch cifar10 fp16 mixed-precision mixed-precision-training

• Updated Oct 25, 2019
• Python

Pytorch implementation of the paper Mixed Precision DNNs: All you need is a good parametrization.

quantization mixed-precision-training

• Updated Feb 8, 2024
• Python

[CVPR 2015] FaceNet: A Unified Embedding for Face Recognition and Clustering

pytorch facenet multi-gpu mixed-precision-training zcls

• Updated Dec 13, 2021
• Python

[TMLR, 2024] Modular Quantization-Aware Training for 6D Object Pose Estimation

model-compression 6dof-pose mixed-precision-training quantization-aware-training

• Updated Jul 27, 2026
• Python

Steel Defect Detection using U-Net. Optimising training and inference using Automatic Mixed Precision and TensorRT respectively.

pytorch kaggle deeplearning tensorrt unet-pytorch mixed-precision-training torch-onnx-trt

• Updated Jun 6, 2020
• HTML

Code repository for "Reducing Underflow in Mixed Precision Training by Gradient Scaling" presented at IJCAI '20

computer-vision deep-learning chainer chainercv mixed-precision-training ijcai2020

• Updated Oct 7, 2021
• Python

demo for pytorch-distributed

amp pytorch distributed mixed-precision-training distributed-data-parallel distributeddataparallel hybrid-precision-training

• Updated Feb 9, 2022
• Python

This repository contains the code and reports for the course INFR11132 Machine Learning Practical. Overall Mark Achieved - 75%

computer-vision deep-learning cnn pytorch resnet vggnet vgg16 mixed-precision-training efficientnet cspnet efficientnetv2

• Updated Jan 4, 2023
• Jupyter Notebook

A Tiny Version of the Original ultralytics/yolov5

python opencv pytorch yolo mosaic onnx yolov3 mixed-precision-training yolov5 distributed-data-parallel

• Updated May 5, 2025
• Python

This project implements a neural network-based chess AI using TensorFlow and Keras. The model uses convolutional layers and residual blocks to predict the best chess moves and evaluate board states. It combines policy and value predictions to create a robust chess-playing AI, inspired by AlphaZero's architecture.

python deep-learning tensorflow neural-networks residual-networks chess-ai mixed-precision-training data-generators reinforcement-learning-concepts fen-parsing board-representation-in-machine-learning

• Updated Nov 20, 2024
• Python

A food vision app is an image classification app for 101 dishes demonstrating the power of transfer learning

tensorflow image-classification transfer-learning fine-tuning mixed-precision-training chatgpt-api

• Updated Apr 3, 2023
• Jupyter Notebook