Universal and customizable implementation of the Hierarchical Variational Autoencoder architecture.
Project description
hVAE - backbone
This repository contains:
- customizable backbone implementation of hVAE
- reuseable hVAE components such as blocks, layers, losses, etc.
- training, evaluation and analyzation scripts
- checkpoint handling (using Weights & Biases)
Installation
pip install hvae_backbone
Usage
This repository is intended to be used as a backend package.
Please refer to hvae template repository for usage instructions.
Project Structure
├── hvae_backbone
│ ├── elements
│ │ ├── __init__.py
│ │ ├── data_preproc.py # Modules for data preprocessing
│ │ ├── dataset.py # Base dataset class
│ │ ├── distributions.py # Distributions, distributions generation
│ │ ├── layers.py # Layers for building models
│ │ ├── losses.py # Loss functions
│ │ ├── nets.py # Network architectures
│ │ ├── optimizers.py # Optimizers
│ │ ├── schedules.py # Schedules e.g. LR, KL weight
│ ├── __init__.py # package level scripts
│ ├── analysis.py # Analysis tools for trained models
│ ├── block.py # Blocks for building hierarchical models
│ ├── checkpoint.py # Checkpoint handling (save, load)
│ ├── functional.py # Functional scripts (training, loss, etc.)
│ ├── hvae.py # General hVAE class
│ ├── sequence.py # General sequential hVAE class
│ ├── utils.py # Utility functions
TODO:
- callbacks
- preprocessing
- sample vs rsmaple blokkoknál (SimpleGenBlock)
- weight initialization
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
hvae_backbone-0.1.116.tar.gz
(34.4 kB
view hashes)
Built Distribution
Close
Hashes for hvae_backbone-0.1.116-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | bf873396dcb234f7a97da2d2b27520f01e51695f1e173a6474bdaeba9eaad91f |
|
MD5 | e5a03e4a762fcb7c80774e5a7adfb4ee |
|
BLAKE2b-256 | 9ed486816fd15ec086f0422fff5c04867b1eed0fd6b8be6388dc973fbf2b57eb |