https://github.com/alisiahkoohi/invertible-deep-prior

invertible-deep-prior
https://github.com/alisiahkoohi/invertible-deep-prior

Last synced: 5 days ago
JSON representation

invertible-deep-prior

• Host: GitHub
• URL: https://github.com/alisiahkoohi/invertible-deep-prior
• Owner: alisiahkoohi
• License: mit
• Created: 2021-04-16T23:08:32.000Z (over 3 years ago)
• Default Branch: main
• Last Pushed: 2021-10-13T02:44:20.000Z (about 3 years ago)
• Last Synced: 2024-10-30T06:05:39.716Z (about 2 months ago)
• Language: Python
• Size: 14.6 KB
• Stars: 1
• Watchers: 3
• Forks: 0
• Open Issues: 0
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# Invertible deep priors

## Prerequisites

First clone the repository:

```bash

git clone https://github.com/alisiahkoohi/invertible-deep-prior

cd invertible-deep-prior/

```

This software is based on [PyTorch-1.8.1](https://github.com/pytorch/pytorch/releases/tag/v1.8.0) and [FrEIA-0.2](https://github.com/VLL-HD/FrEIA/releases/tag/v0.2).

Follow the steps below to install the necessary libraries. If you have a CUDA-enabled GPU, run:

```bash

conda env create -f environment.yml

source activate freia

```

You may want to specify your CUDA Toolkit version in the `environment.yml` file above. If you don't have a GPU, run:

```bash

conda env create -f environment-cpu.yml

source activate freia-cpu

```

## Example

To run the example, execute:

```bash

python src/main.py --cuda 1 --lr 0.0001 --max_itr 5001

```

## Author

Ali Siahkoohi