From db255e7c38e13568f2b25134a6679c6ae2a0a1de Mon Sep 17 00:00:00 2001 From: victolee0 <39608452+victolee0@users.noreply.github.com> Date: Tue, 15 Mar 2022 09:27:05 +0900 Subject: [PATCH] fix vnet typo fix docstring typo Signed-off-by: victolee0 <39608452+victolee0@users.noreply.github.com> --- monai/networks/nets/vnet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monai/networks/nets/vnet.py b/monai/networks/nets/vnet.py index ede5f688aa..7669b4678e 100644 --- a/monai/networks/nets/vnet.py +++ b/monai/networks/nets/vnet.py @@ -213,7 +213,7 @@ class VNet(nn.Module): The value should meet the condition that ``16 % in_channels == 0``. out_channels: number of output channels for the network. Defaults to 1. act: activation type in the network. Defaults to ``("elu", {"inplace": True})``. - dropout_prob: dropout ratio. Defaults to 0.5. Defaults to 3. + dropout_prob: dropout ratio. Defaults to 0.5. dropout_dim: determine the dimensions of dropout. Defaults to 3. - ``dropout_dim = 1``, randomly zeroes some of the elements for each channel.