Neural Network Compression Framework for enhanced OpenVINO™ inference

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

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

Distributed, mixed-precision training with PyTorch

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

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

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

Using Deep Learning To Identify And Classify Building Damage

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

demo for pytorch-distributed

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

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

A Tiny Version of the Original ultralytics/yolov5

This repository contains a Convolutional Neural Network (CNN) model designed for brain tumor classification using MRI images. The model employs multiple convolutional layers, batch normalization, dropout for regularization, and fully connected layers to achieve high accuracy.

[CVPR 2017]YOLO9000: Better, Faster, Stronger

This application predicts the name of a country (or countries) based on an input flag image. It uses advanced image processing techniques and deep learning models built with PyTorch to classify flags accurately.