PyTorch code accompanying our paper on Maximum Entropy Generators for Energy-Based Models

Last update: Oct 27, 2022

Overview

Maximum Entropy Generators for Energy-Based Models

All experiments have tensorboard visualizations for samples / density / train curves etc.

To run the toy data experiments:

python scripts/train/ebm_toy.py --dataset swissroll --save_path logs/swissroll

To run the discrete mode collapse experiment:

python scripts/train/ebm_mnist.py --save_path logs/mnist_3 --n_stack 3

This requires the pretrained mnist classifier:

python scripts/train/mnist_classifier.py

To run the CIFAR image generation experiment:

python scripts/train/ebm_cifar.py --save_path logs/cifar

To run the MCMC evalulations on CIFAR data:

python scripts/test/eval_metrics_cifar --load_path logs/cifar --n_samples 50000 --mcmc_iters 5 --temp .01

NOTE: This requires cloning the TTUR repo in the current working directory (https://github.com/bioinf-jku/TTUR).

To run the CelebA image generation experiments:

python scripts/train/ebm_celeba.py --save_path logs/celeba

NOTE: Results are subject to PyTorch version. I have already noticed variance in quantitative numbers with PyTorch version upgrades.

PyTorch code accompanying our paper on Maximum Entropy Generators for Energy-Based Models

Related tags

Overview

Maximum Entropy Generators for Energy-Based Models

Owner

Rithesh Kumar

AISTATS 2019: Confidence-based Graph Convolutional Networks for Semi-Supervised Learning

PASTRIE: A Corpus of Prepositions Annotated with Supersense Tags in Reddit International English

Official PyTorch implementation of paper: Standardized Max Logits: A Simple yet Effective Approach for Identifying Unexpected Road Obstacles in Urban-Scene Segmentation (ICCV 2021 Oral Presentation)

Atomistic Line Graph Neural Network

Utility code for use with PyXLL

OpenDILab Multi-Agent Environment

This repository contains code to train and render Mixture of Volumetric Primitives (MVP) models

RL-GAN: Transfer Learning for Related Reinforcement Learning Tasks via Image-to-Image Translation

PyTorch implementation of the Transformer in Post-LN (Post-LayerNorm) and Pre-LN (Pre-LayerNorm).

magiCARP: Contrastive Authoring+Reviewing Pretraining

Code for "OctField: Hierarchical Implicit Functions for 3D Modeling (NeurIPS 2021)"

LeafSnap replicated using deep neural networks to test accuracy compared to traditional computer vision methods.

A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".

code for paper "Not All Unlabeled Data are Equal: Learning to Weight Data in Semi-supervised Learning" by Zhongzheng Ren, Raymond A. Yeh, Alexander G. Schwing.

ScriptProfilerPy - Module to visualize where your python script is slow

Unofficial pytorch implementation of the paper "Dynamic High-Pass Filtering and Multi-Spectral Attention for Image Super-Resolution"

AlgoVision - A Framework for Differentiable Algorithms and Algorithmic Supervision

Cereal box identification in store shelves using computer vision and a single train image per model.

Get a Grip! - A robotic system for remote clinical environments.

Official PyTorch Implementation of GAN-Supervised Dense Visual Alignment

PyTorch code accompanying our paper on Maximum Entropy Generators for Energy-Based Models

Related tags

Overview

Maximum Entropy Generators for Energy-Based Models

Owner

Rithesh Kumar

AISTATS 2019: Confidence-based Graph Convolutional Networks for Semi-Supervised Learning

PASTRIE: A Corpus of Prepositions Annotated with Supersense Tags in Reddit International English

Official PyTorch implementation of paper: Standardized Max Logits: A Simple yet Effective Approach for Identifying Unexpected Road Obstacles in Urban-Scene Segmentation (ICCV 2021 Oral Presentation)

Atomistic Line Graph Neural Network

Utility code for use with PyXLL

OpenDILab Multi-Agent Environment

This repository contains code to train and render Mixture of Volumetric Primitives (MVP) models

RL-GAN: Transfer Learning for Related Reinforcement Learning Tasks via Image-to-Image Translation

PyTorch implementation of the Transformer in Post-LN (Post-LayerNorm) and Pre-LN (Pre-LayerNorm).

magiCARP: Contrastive Authoring+Reviewing Pretraining

Code for "OctField: Hierarchical Implicit Functions for 3D Modeling (NeurIPS 2021)"

LeafSnap replicated using deep neural networks to test accuracy compared to traditional computer vision methods.

A PyTorch implementation of Mugs proposed by our paper "Mugs: A Multi-Granular Self-Supervised Learning Framework".

code for paper "Not All Unlabeled Data are Equal: Learning to Weight Data in Semi-supervised Learning" by Zhongzheng Ren*, Raymond A. Yeh*, Alexander G. Schwing.

ScriptProfilerPy - Module to visualize where your python script is slow

Unofficial pytorch implementation of the paper "Dynamic High-Pass Filtering and Multi-Spectral Attention for Image Super-Resolution"

AlgoVision - A Framework for Differentiable Algorithms and Algorithmic Supervision

Cereal box identification in store shelves using computer vision and a single train image per model.

Get a Grip! - A robotic system for remote clinical environments.

Official PyTorch Implementation of GAN-Supervised Dense Visual Alignment

code for paper "Not All Unlabeled Data are Equal: Learning to Weight Data in Semi-supervised Learning" by Zhongzheng Ren, Raymond A. Yeh, Alexander G. Schwing.