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.53.tar.gz
(33.7 kB
view hashes)
Built Distribution
Close
Hashes for hvae_backbone-0.1.53-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b44e3d9da19709a390268e0d5a2abe04ecfa9549248de506476f30d64bc15000 |
|
MD5 | c6138ef6a1016eeb5922c20ddd59d680 |
|
BLAKE2b-256 | d51efe5f1274889f41f40639be54c4c2e57ebf8108e7a89bacc1abc0907fac9c |