File tree Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Expand file tree Collapse file tree 1 file changed +1
-1
lines changed Original file line number Diff line number Diff 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.
You can’t perform that action at this time.
0 commit comments