Gwena Cunha

I have the same question. Actually, are you doing over/under-sampling or just re-weighting the samples in the training stage? Is there a paper that I can refer to?

@daverim is there any update on this? I saw that there's a [new blog from TensorFlow](https://blog.tensorflow.org/2022/06/Adding-Quantization-aware-Training-and-Pruning-to-the-TensorFlow-Model-Garden.html) saying that Subclassed models are now supported, but not much information is given on...

Thank you for the reply @Xhark! Is there an ETA on when Subclassed models will be supported? Thanks!

Any updates on this?

@daverim I encountered the same error log for SeparableConv2D using TF 2.8.0 (no error with DepthwiseConv2D in that TF version): ``` ... Traceback (most recent call last): File "/home/PycharmProjects/tensorrt_qat/examples/mobilenet/run_qat_workflow.py", line...

Hi @k-w-w thank you for your feedback! This specific issue (for DepthwiseConv) has been solved, as mentioned in a comment on Jan 26th above, but the same issue persists for...

@k-w-w yes

@k-w-w : ``` ... Traceback (most recent call last): File "/home/nvidia/PycharmProjects/nvbugs/internal_filed/tf_key_inference_bug/TF_bug_separableconv2d/sample.py", line 24, in model.save(model_save_path) File "/home/nvidia/PycharmProjects/nvbugs/venv38_trt_regression/lib/python3.8/site-packages/keras/utils/traceback_utils.py", line 67, in error_handler raise e.with_traceback(filtered_tb) from None File "/home/nvidia/PycharmProjects/nvbugs/venv38_trt_regression/lib/python3.8/site-packages/tensorflow/python/saved_model/save.py", line 403, in...

[This bug](https://github.com/tensorflow/model-optimization/issues/964) also has the reproducible code, so we can move our discussion there if you agree.

This bug **can be closed** for `DepthwiseConv2D`. For `Conv2DTranspose` and `SeparableConv2D`, please move the discussion [here](https://github.com/tensorflow/model-optimization/issues/964). Thank you!