model export is taking forever and sometimes using full RAM resulting in crashing..

googlecodelabs / odml-pathways

Apache License 2.0

47 stars 31 forks source link

model export is taking forever and sometimes using full RAM resulting in crashing.. #4

Closed ManoharSai2000 closed 3 years ago

ManoharSai2000 commented 3 years ago

Expected Behavior

model.export() should give a tflite file output.

Actual Behavior

It is taking forever to run, and sometimes it crashes.

Steps to Reproduce the Problem

Run the notebook but with effcient det lite 4 version.

Specifications

Version: Colab up to date
Platform: Colab

Update:

It took 4.5 hrs for converting the model!. But again the evaluate_tflite is only evaluating on part of the test set and stopping!

khanhlvg commented 3 years ago

This seems to be the resource limitation of the free Colab. I couldn't reproduce the issue on Colab Pro's High-RAM version. Please try again using a machine with higher specs (either Colab Pro, or use GCP's VM, or your local machine)