/FSQ

Keras implement of Finite Scalar Quantization

Primary LanguagePythonApache License 2.0Apache-2.0

FSQ

Keras implement of Finite Scalar Quantization

Introduction

Code

  • fsqae.py: Finite Scalar Quantised - AutoEncoder
  • vqvae.py: Vector Quantised - Variational AutoEncoder

Environment

  • tensorflow 1.15.5
  • keras 2.3.1
  • bert4keras 0.11.5