jonbarron / robust_loss_pytorch
A pytorch port of google-research/google-research/robust_loss/
☆658Updated 3 years ago
Related projects: ⓘ
- Geometric loss functions between point clouds, images and volumes☆582Updated 8 months ago
- Official Code for Invertible Residual Networks☆517Updated last month
- Approximating Wasserstein distances with PyTorch☆452Updated last year
- Code snippets created for the PyTorch discussion board☆539Updated 3 years ago
- Totally Versatile Miscellanea for Pytorch☆461Updated 2 years ago
- Tiny PyTorch library for maintaining a moving average of a collection of parameters.☆400Updated 2 years ago
- Implementing Stand-Alone Self-Attention in Vision Models using Pytorch☆454Updated 4 years ago
- [NeurIPS'19] Deep Equilibrium Models☆719Updated 2 years ago
- A PyTorch implementation of the 1d and 2d Sinusoidal positional encoding/embedding.☆251Updated 3 years ago
- Toolbox to integrate optimal transport loss functions using automatic differentiation and Sinkhorn's algorithm☆426Updated 6 years ago
- The Official PyTorch Implementation of "NVAE: A Deep Hierarchical Variational Autoencoder" (NeurIPS 2020 spotlight paper)☆1,012Updated last year
- Gaussian mixture models in PyTorch.☆501Updated last year
- PyTorch implementation of Sinusodial Representation networks (SIREN)☆263Updated last year
- Gradually-Warmup Learning Rate Scheduler for PyTorch☆971Updated 3 years ago
- Compute CNN receptive field size in pytorch in one line☆348Updated 4 months ago
- Framework for Easily Invertible Architectures☆756Updated 2 months ago
- Pytorch implementation of SIREN - Implicit Neural Representations with Periodic Activation Function☆465Updated last year
- [arXiv 2019] "Contrastive Multiview Coding", also contains implementations for MoCo and InstDis☆1,299Updated 3 years ago
- [NeurIPS'20] Multiscale Deep Equilibrium Models☆232Updated 3 years ago
- Fourier Features Let Networks Learn High Frequency Functions in Low Dimensional Domains☆1,232Updated last year
- [CVPR 2020 Workshop] A PyTorch GAN library that reproduces research results for popular GANs.☆601Updated 2 years ago
- PyTorch Boilerplate For Research☆603Updated 2 years ago
- Standalone TFRecord reader/writer with PyTorch data loaders☆855Updated 3 weeks ago
- Implementation of LambdaNetworks, a new approach to image recognition that reaches SOTA with less compute☆1,530Updated 3 years ago
- Simple, extendable, easy to understand Glow implementation in PyTorch☆374Updated 2 years ago
- Pytorch implementations of density estimation algorithms: BNAF, Glow, MAF, RealNVP, planar flows☆599Updated 3 years ago
- My best practice of training large dataset using PyTorch.☆1,080Updated 4 months ago
- Implements https://arxiv.org/abs/1711.05101 AdamW optimizer, cosine learning rate scheduler and "Cyclical Learning Rates for Training Neu…☆148Updated 5 years ago
- Unofficial implementation of: Multi-task learning using uncertainty to weigh losses for scene geometry and semantics☆537Updated 2 years ago
- An All-MLP solution for Vision, from Google AI☆987Updated this week