Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

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

Open
kissablemt opened this issue Jan 14, 2022 · 0 comments

Comments

@kissablemt
Copy link

kissablemt commented Jan 14, 2022

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")加载模型
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant