updates

hudes/param_nn.py

@@ -24,6 +24,7 @@ def __init__(self, input_channels, output_channels):
         self.bias_params = self.output_channels
         self.params = self.weight_params + self.bias_params
 
+    # @torch.compile
     def forward(self, models_params, x):
         # model_params ~ (models, params)
         _weights = models_params[:, : self.weight_params].reshape(