Add experimental softplus to LFQ (#97)

avramdj · web-flow · commit 289767d63fa5 · 2024-12-04T06:09:13.000-08:00
diff --git a/meshgpt_pytorch/meshgpt_pytorch.py b/meshgpt_pytorch/meshgpt_pytorch.py
@@ -457,6 +457,7 @@ def __init__(
             quantize_dropout = True,
             quantize_dropout_cutoff_index = 1,
             quantize_dropout_multiple_of = 1,
+            experimental_softplus_entropy_loss=True,
         ),
         rvq_kwargs: dict = dict(
             kmeans_init = True,
