adityaiitb / pyprof2
PyProf2: PyTorch Profiling tool
☆82Updated 4 years ago
Alternatives and similar repositories for pyprof2:
Users that are interested in pyprof2 are comparing it to the libraries listed below
- This is a PyTorch implementation of the Scalpel. Node pruning for five benchmark networks and SIMD-aware weight pruning for LeNet-300-100…☆41Updated 6 years ago
- Code for paper "SWALP: Stochastic Weight Averaging forLow-Precision Training".☆62Updated 5 years ago
- ☆53Updated 7 years ago
- Example repository for custom C++/CUDA operators for TorchScript☆114Updated 2 years ago
- Customized matrix multiplication kernels☆54Updated 3 years ago
- "Layer-wise Adaptive Rate Scaling" in PyTorch☆86Updated 4 years ago
- Programmable Neural Network Compression☆148Updated 2 years ago
- Use TensorFlow efficiently☆95Updated 3 years ago
- PyTorch implementation of Wide Residual Networks with 1-bit weights by McDonnell (ICLR 2018)☆125Updated 6 years ago
- Filter Response Normalization tested on better ImageNet baselines.☆35Updated 5 years ago
- Simple Training and Deployment of Fast End-to-End Binary Networks☆157Updated 3 years ago
- Training neural networks with 8-bit computations☆28Updated 9 years ago
- MONeT framework for reducing memory consumption of DNN training☆173Updated 4 years ago
- Train ImageNet in 18 minutes on AWS☆129Updated last year
- DLPack for Tensorflow☆35Updated 5 years ago
- Simple implementation of the LSUV initialization in PyTorch☆58Updated last year
- Efficient DataLoader for PyTorch and Keras for loading datasets from web servers and object stores.☆29Updated 5 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆131Updated 3 years ago
- Efficient forward propagation for BCNNs☆50Updated 7 years ago
- Low Precision Arithmetic Simulation in PyTorch☆276Updated 11 months ago
- A PyTorch implementation of "Incremental Network Quantization: Towards Lossless CNNs with Low-Precision Weights"☆167Updated 5 years ago
- Fast matrix multiplication for few-bit integer matrices on CPUs.☆27Updated 6 years ago
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- ☆47Updated 5 years ago
- Repository containing pruned models and related information☆37Updated 4 years ago
- an example of a CUDA extension for PyTorch using CuPy which computes the Hadamard product of two tensors☆118Updated 4 months ago
- How and why you want to make your pytorch CUDA/CPP extension with a Makefile☆172Updated 5 years ago
- Profile the GPU memory usage of every line in a Pytorch code☆82Updated 6 years ago
- Example of backprop which uses constant memory☆43Updated 6 years ago
- Distributed, mixed-precision training with PyTorch☆90Updated 4 years ago