Energy-OOD

This repo is a PyTorch implementation of the paper Energy-based Out-of-distribution Detection

Installation

Download the Tiny-images dataset and put them under data folder
Setup environment.

Assuming PyTorch is already installed.

pip install tensorboard ood-metrics

Run python main.py to train a model using CIFAR10 as in-distribution dataset and SVHN as out-of-distribution dataset.

Name		Name	Last commit message	Last commit date
Latest commit History 4 Commits
test		test
README.md		README.md
dataloader.py		dataloader.py
main.py		main.py
model.py		model.py
trainer.py		trainer.py