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]`
|
||||
representative_dataset: A representative dataset that can be used to
|
||||
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.
|
||||
Enables MLIR-based conversion instead of TOCO conversion.
|
||||
experimental_new_quantizer: Experimental flag, subject to change.
|
||||
|
@ -80,7 +80,7 @@ class Nadam(optimizer_v2.OptimizerV2):
|
||||
rate for the exponentially weighted infinity norm.
|
||||
epsilon: A small constant for numerical stability.
|
||||
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`,
|
||||
`decay`}. `clipnorm` is clip gradients by norm; `clipvalue` is clip
|
||||
gradients by value, `decay` is included for backward compatibility to
|
||||
|
Loading…
Reference in New Issue
Block a user