tomgoldstein / loss-landscape
Code for visualizing the loss landscape of neural nets
☆2,917Updated 2 years ago
Alternatives and similar repositories for loss-landscape:
Users that are interested in loss-landscape are comparing it to the libraries listed below
- A small package to create visualizations of PyTorch execution graphs☆3,299Updated last month
- torch-optimizer -- collection of optimizers for Pytorch☆3,081Updated 10 months ago
- SAM: Sharpness-Aware Minimization (PyTorch)☆1,829Updated 11 months ago
- Model summary in PyTorch similar to `model.summary()` in Keras☆4,027Updated 11 months ago
- higher is a pytorch library allowing users to obtain higher order gradients over losses spanning training loops rather than individual tr…☆1,606Updated 2 years ago
- Count the MACs / FLOPs of your PyTorch model.☆4,955Updated 7 months ago
- Neural network graphs and training metrics for PyTorch, Tensorflow, and Keras.☆1,810Updated last year
- tensorboard for pytorch (and chainer, mxnet, numpy, ...)☆7,913Updated last month
- AI conference deadline countdowns☆5,756Updated 5 months ago
- GPU Accelerated t-SNE for CUDA with Python bindings☆1,824Updated 4 months ago
- Model interpretability and understanding for PyTorch☆5,080Updated this week
- A lightweight library for PyTorch training tools and utilities☆1,685Updated this week
- Codebase for Image Classification Research, written in PyTorch.☆2,148Updated 10 months ago
- A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch☆8,537Updated 2 weeks ago
- Pytorch implementation of convolutional neural network visualization techniques☆7,948Updated last month
- Pytorch library for fast transformer implementations☆1,677Updated last year
- Collection of common code that's shared among different research projects in FAIR computer vision team.☆2,077Updated 2 months ago
- Tips for releasing research code in Machine Learning (with official NeurIPS 2020 recommendations)☆2,619Updated last year
- Flexible and powerful tensor operations for readable and reliable code (for pytorch, jax, TF and others)☆8,741Updated last week
- The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.☆6,088Updated 2 months ago
- Unofficial implementation of the ImageNet, CIFAR 10 and SVHN Augmentation Policies learned by AutoAugment using pillow☆1,478Updated last year
- Model analyzer in PyTorch☆1,476Updated last year
- Ranger - a synergistic optimizer using RAdam (Rectified Adam), Gradient Centralization and LookAhead in one codebase☆1,196Updated last year
- Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch☆1,794Updated 7 months ago
- PyTorch implementation of "Supervised Contrastive Learning" (and SimCLR incidentally)☆3,191Updated last year
- Train AI models efficiently on medical images using any framework☆1,870Updated 8 months ago
- A Collection of Variational Autoencoders (VAE) in PyTorch.☆6,914Updated 8 months ago
- Official Implementation of 'Fast AutoAugment' in PyTorch.☆1,602Updated 3 years ago
- PyTorch implementation of MoCo: https://arxiv.org/abs/1911.05722☆4,885Updated 2 months ago
- PyTorch implementation of Contrastive Learning methods☆1,958Updated last year