Non Square SSDLITE MOBILEDET model fails to compile in edge tpu compiler

[KushalHM]

Below is the compiler error:

Edge TPU Compiler version 16.0.384591198 Started a compilation timeout timer of 180 seconds. ERROR: :317 scale_diff / output_scale <= 0.02 was not true. ERROR: Node number 3 (CONV_2D) failed to prepare.

Compilation failed: Model failed in Tflite interpreter. Please ensure model can be loaded/run in Tflite interpreter. Compilation child process completed within timeout period. Compilation failed!

Is there are special operations to be performed while training or while exporting to get past this error?

NOTE: same training pipeline with a square fixed image resize / resolution (ex: 480x480), on the custom dataset compiles, loads and executes inside the edge tpu.