diff --git a/monai/networks/nets/densenet.py b/monai/networks/nets/densenet.py index 2ec6efcb44..0d1aa8c447 100644 --- a/monai/networks/nets/densenet.py +++ b/monai/networks/nets/densenet.py @@ -174,7 +174,7 @@ def __init__( ("relu", nn.ReLU(inplace=True)), ("norm", avg_pool_type(1)), ("flatten", nn.Flatten(1)), - ("class", nn.Linear(in_channels, out_channels)), + ("out", nn.Linear(in_channels, out_channels)), ] ) )