Skip to content

Commit 4fc1ea9

Browse files
committed
clarify default for channels last conversion in pytorch
1 parent 7cf4134 commit 4fc1ea9

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

hls4ml/utils/config.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -318,7 +318,7 @@ def config_from_pytorch_model(
318318
be an explicit precision: 'auto' is not allowed.
319319
default_reuse_factor (int, optional): Default reuse factor. Defaults to 1.
320320
channels_last_conversion (string, optional): Configures the conversion of pytorch layers to
321-
'channels_last' data format used by hls4ml internally. Can be set to 'full', 'internal',
321+
'channels_last' data format used by hls4ml internally. Can be set to 'full' (default), 'internal',
322322
or 'off'. If 'full', both the inputs and internal layers will be converted. If 'internal',
323323
only internal layers will be converted; this assumes the inputs are converted by the user.
324324
If 'off', no conversion is performed.

0 commit comments

Comments
 (0)