libcoral icon indicating copy to clipboard operation
libcoral copied to clipboard
verified publisher icon google-coral

Transfer Learning with back propagation with tfilte model

Open abucka opened this issue 4 years ago • 3 comments

Description

Is it possible to also retrain a pure tflite model without edgetpu using back propagation method?

Click to expand!

Issue Type

Support

Operating System

Linux

Coral Device

No response

Other Devices

No response

Programming Language

C++

Relevant Log Output

No response

abucka avatar Mar 02 '22 08:03 abucka

Hello @abucka I think, it is possible. Please feel free to try to retrain pure tflite models from test_data repo using back propagation method.

hjonnala avatar Mar 02 '22 18:03 hjonnala

Hi, I tested it but unfortunately the training process hangs and does not output any information about loss or accuracy like it happens in case of edgetpu version.

abucka avatar Mar 07 '22 11:03 abucka

The process might be hanging due to CPU limitation. Can you try with one or two images from each dateset?

hjonnala avatar Mar 08 '22 18:03 hjonnala