lucidrains / vector-quantize-pytorch

Vector (and Scalar) Quantization, in Pytorch
MIT License
2.12k stars 179 forks source link

Small refacto #138

Closed MisterBourbaki closed 1 day ago

MisterBourbaki commented 1 week ago

This PR proposes a first step towards a better organization of the code, by adding a utils module which contains all basic function used throughout the library.

Add: a utils module

Fix: in LFQ, replaces the use of lambda function by simply using the 2**dim variable.

Lint: all imports in all files.