転移学習をしたTesnorflow lite がAndroid Studio上で動かない

Question

やりたいこと

こちらの記事を参考にtensorflowで転移学習を用いた学習モデルをAndroid Studioで動かしたい。

追記
こちらのイシューにあったように学習回数がゼロのモデルを使用したところ動くことが確認できた。しかし一回でも学習をすると下のエラーが発生してしまった。同じようなエラーを経験した方などの解決法または別の方法を教えていただきたいです。
環境
windows 10
Android Studio 4.01
Tensorflow 1.15
Tensorflowモデル　ssd_mobilenet_v3_large_coco

エラーと発生個所

java.lang.IllegalArgumentException: Internal error: Failed to run on the given Interpreter: tensorflow/lite/kernels/detection_postprocess.cc:426 ValidateBoxes(decoded_boxes, num_boxes) was not true.
Node number 180 (TFLite_Detection_PostProcess) failed to invoke.

at　com.example.TFLiteObjectDetectionAPIModel.recognizeImage　(TFLiteObjectDetectionAPIModel.java:223)
at com.example.SettingActivity$4.onClick(SettingActivity.java:334)

　final List<Detector.Recognition> results = detector.recognizeImage(crop);

    tfLite.runForMultipleInputsOutputs(inputArray, outputMap);

試したこと

こちらに同じエラーのイシューがあるのは見つけたのですがssdlite_mobilnet_v2の場合のみ起こるようでssd_mobilenet_v3_large_cocoの場合の対処法が分からなかった。

Tensorflow lite を用いるにあたってバージョンの問題でこれまでも様々なところでつまづいたため何か少しでもわかることがありましたらよろしくお願いいたします。