Merge pull request #37192 from ashutosh1919:tflite_piq
PiperOrigin-RevId: 301442417 Change-Id: Iec4bba1d74354ec0d873ff8477550021602e7aec
This commit is contained in:
commit
83c0affbbd
@ -365,7 +365,9 @@ class TFLiteConverterV2(TFLiteConverterBase):
|
|||||||
to apply when converting the model. E.g. `[Optimize.DEFAULT]`
|
to apply when converting the model. E.g. `[Optimize.DEFAULT]`
|
||||||
representative_dataset: A representative dataset that can be used to
|
representative_dataset: A representative dataset that can be used to
|
||||||
generate input and output samples for the model. The converter can use the
|
generate input and output samples for the model. The converter can use the
|
||||||
dataset to evaluate different optimizations.
|
dataset to evaluate different optimizations. Note that this is an optional
|
||||||
|
attribute but it is necessary if INT8 is the only support builtin ops in
|
||||||
|
target ops.
|
||||||
experimental_new_converter: Experimental flag, subject to change.
|
experimental_new_converter: Experimental flag, subject to change.
|
||||||
Enables MLIR-based conversion instead of TOCO conversion.
|
Enables MLIR-based conversion instead of TOCO conversion.
|
||||||
experimental_new_quantizer: Experimental flag, subject to change.
|
experimental_new_quantizer: Experimental flag, subject to change.
|
||||||
|
@ -80,7 +80,7 @@ class Nadam(optimizer_v2.OptimizerV2):
|
|||||||
rate for the exponentially weighted infinity norm.
|
rate for the exponentially weighted infinity norm.
|
||||||
epsilon: A small constant for numerical stability.
|
epsilon: A small constant for numerical stability.
|
||||||
name: Optional name for the operations created when applying gradients.
|
name: Optional name for the operations created when applying gradients.
|
||||||
Defaults to "Adamax".
|
Defaults to "Nadam".
|
||||||
**kwargs: keyword arguments. Allowed to be {`clipnorm`, `clipvalue`, `lr`,
|
**kwargs: keyword arguments. Allowed to be {`clipnorm`, `clipvalue`, `lr`,
|
||||||
`decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip
|
`decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip
|
||||||
gradients by value, `decay` is included for backward compatibility to
|
gradients by value, `decay` is included for backward compatibility to
|
||||||
|
Loading…
Reference in New Issue
Block a user