We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8a936d2 commit bc423a5Copy full SHA for bc423a5
1 file changed
optimum/exporters/tflite/base.py
@@ -169,10 +169,10 @@ def __init__(
169
point_batch_size: Optional[int] = None,
170
nb_points_per_image: Optional[int] = None,
171
):
172
- super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
173
self.mandatory_axes = ()
174
self._axes: Dict[str, int] = {}
175
- self.task = task
+
+ super().__init__(config=config, task=task, int_dtype="int64", float_dtype="fp32")
176
177
# To avoid using **kwargs.
178
axes_values = {
0 commit comments