sageconv_kwargs

Author: MarcusLoppe

meshgpt_pytorch/meshgpt_pytorch.py

@@ -485,7 +485,7 @@ def __init__(
         init_encoder_dim, *encoder_dims_through_depth = encoder_dims_through_depth
         curr_dim = init_encoder_dim
 
-        self.init_sage_conv = SAGEConv(dim_codebook, init_encoder_dim, {**sageconv_kwargs})
+        self.init_sage_conv = SAGEConv(dim_codebook, init_encoder_dim, **sageconv_kwargs)
 
         self.init_encoder_act_and_norm = nn.Sequential(
             nn.SiLU(),