Unverified Commit cb63febf authored by George Ogden's avatar George Ogden Committed by GitHub
Browse files

Update documentation (#2996)

* Update documentation

Based on sampling, the width and height must be powers of 2 as the samples halve in size each time

* make style
parent 8c6b47cf
......@@ -44,7 +44,8 @@ class UNet2DModel(ModelMixin, ConfigMixin):
Parameters:
sample_size (`int` or `Tuple[int, int]`, *optional*, defaults to `None`):
Height and width of input/output sample.
Height and width of input/output sample. Dimensions must be a multiple of `2 ** (len(block_out_channels) -
1)`.
in_channels (`int`, *optional*, defaults to 3): Number of channels in the input image.
out_channels (`int`, *optional*, defaults to 3): Number of channels in the output.
center_input_sample (`bool`, *optional*, defaults to `False`): Whether to center the input sample.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment