lucidrains / vector-quantize-pytorch

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

Small refacto #138

Closed MisterBourbaki closed 4 months ago

MisterBourbaki commented 5 months 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.