Fix conv fc model (#98)

* make positional args explicit
NVIDIA · Dec 19, 2023 · 357b119 · 357b119
1 parent 9d056fc
commit 357b119
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 5 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -18,6 +18,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ### Fixed
 
+- Fix bug for `ConvFullyConnectedArch`.
+
 ### Security
 
 ### Dependencies

diff --git a/modulus/sym/models/fully_connected.py b/modulus/sym/models/fully_connected.py
@@ -66,11 +66,13 @@ def __init__(
         for i in range(nr_layers):
             self.layers.append(
                 fc_layer(
-                    layer_in_features,
-                    layer_size,
-                    get_activation_fn(activation_fn[i], out_features=out_features),
-                    weight_norm,
-                    activation_par,
+                    in_features=layer_in_features,
+                    out_features=layer_size,
+                    activation_fn=get_activation_fn(
+                        activation_fn[i], out_features=out_features
+                    ),
+                    weight_norm=weight_norm,
+                    activation_par=activation_par,
                 )
             )
             layer_in_features = layer_size