From 4e8b327c22dfac83d832ad1526907d2fb855468f Mon Sep 17 00:00:00 2001 From: Jorge Prado Date: Mon, 2 Sep 2024 11:55:07 +0200 Subject: [PATCH] Fix Style issues --- src/graphnet/models/gnn/particlenet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/graphnet/models/gnn/particlenet.py b/src/graphnet/models/gnn/particlenet.py index 4ff51691b..cf2d00998 100644 --- a/src/graphnet/models/gnn/particlenet.py +++ b/src/graphnet/models/gnn/particlenet.py @@ -72,7 +72,7 @@ def __init__( Default to "mean". activation_layer: The activation function to use in the model. Default to "relu". - add_batchnorm_layer: Whether to add a batch normalization layer + add_batchnorm_layer: Whether to add a batch normalization layer after each linear layer. Default to True. dropout_readout: Dropout value to use in the readout layer(s). Default to 0.1.