2635 Enhance doc-string of UNet for spatial size

[Nic-Ma]

Fixes #2635 .

Description

As several users asked questions about the input size of UNet, this PR enhanced the doc-string to make it clear.

Status

Ready

Types of changes

• Non-breaking change (fix or new feature that would not break existing functionality).
• Breaking change (fix or new feature that would cause existing functionality to change).
• New tests added to cover the changes.
• Integration tests passed locally by running ./runtests.sh -f -u --net --coverage.
• Quick tests passed locally by running ./runtests.sh --quick --unittests.
• In-line docstrings updated.
• Documentation updated, tested make html command in the docs/ folder.

[Nic-Ma]

/black

[Nic-Ma]

/black

[Nic-Ma]

/black

[yiheng-wang-nv]

Supplement: test_unet.py does not cover the parameters: kernel_size and up_kernel_size, and strides is always (2, 2). I think more test cases are necessary.

[Nic-Ma]

Supplement: test_unet.py does not cover the parameters: kernel_size and up_kernel_size, and strides is always (2, 2). I think more test cases are necessary.

Sure, please feel free to submit commits directly.

Thanks.

[Nic-Ma]

/black

[Nic-Ma]

/black