add None in the final_layers_options

benmalef · Feb 25, 2025 · 71ade3f · 71ade3f
1 parent a70c88b
commit 71ade3f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/GANDLF/Configuration/Parameters/model_parameters.py b/GANDLF/Configuration/Parameters/model_parameters.py
@@ -13,7 +13,7 @@
 NORM_TYPE_OPTIONS = Literal["batch", "instance", "none"]
 # Define model final_layer options
 FINAL_LAYER_OPTIONS = Literal[
-    "sigmoid", "softmax", "logsoftmax", "tanh", "identity", "logits", "regression"
+    "sigmoid", "softmax", "logsoftmax", "tanh", "identity", "logits", "regression", "None", "none"
 ]