diff --git a/monai/networks/nets/densenet.py b/monai/networks/nets/densenet.py index 8fb2c269ab..52bd2fa994 100644 --- a/monai/networks/nets/densenet.py +++ b/monai/networks/nets/densenet.py @@ -148,6 +148,9 @@ class DenseNet(nn.Module): """ Densenet based on: `Densely Connected Convolutional Networks `_. Adapted from PyTorch Hub 2D version: https://pytorch.org/vision/stable/models.html#id16. + This network is non-determistic When `spatial_dims` is 3 and CUDA is enabled. Please check the link below + for more details: + https://pytorch.org/docs/stable/generated/torch.use_deterministic_algorithms.html#torch.use_deterministic_algorithms Args: spatial_dims: number of spatial dimensions of the input image.