maix_train icon indicating copy to clipboard operation
maix_train copied to clipboard

The solution of model breakpoint continuous training(断点续训,无需重新训练)

Open kissablemt opened this issue 3 years ago • 0 comments

Take mobilenet's classification model as an example

Remember to backup!

  1. By default, after your training, there is a m.tflite.h5 file in maix_train/out. Copy it to some place else.
  2. You need to modify lines 121 to 129 in maix_train/train/classifier/__init__.py
  3. Load the model with self.model = tf.keras.models.load_model("Your filepath of *.h5 ")

image


Mobilenet的分类模型测试过可行

记得备份!

  1. 默认情况下,在你训练之后,在 maix_train/out 中有一个 m.tflite.h5 文件。 将其复制到其他地方。
  2. 需要修改maix_train/train/classifier/__init__.py中的121到129行
  3. 使用self.model = tf.keras.models.load_model("Your filepath of *.h5")加载模型

kissablemt avatar Jan 14 '22 05:01 kissablemt