qurator-spk/eynollah
1
0
Fork 0
mirror of https://github.com/qurator-spk/eynollah.git synced 2026-03-01 21:02:00 +01:00
Code Issues Projects Releases Wiki Activity

Fix a bug by saving model weights in steps

Browse source
This commit is contained in:
vahidrezanezhad 2026-02-28 09:01:06 +01:00
parent d8667f46d7
commit fed005abd7
1 changed files with 3 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

5
src/eynollah/training/train.py
View file

@ -58,6 +58,7 @@ class SaveWeightsAfterSteps(Callback):
self.save_path = save_path
self.step_count = 0
self._config = _config
self.characters_cnnrnn_ocr = characters_cnnrnn_ocr
def on_train_batch_end(self, batch, logs=None):
self.step_count += 1
@ -68,8 +69,8 @@ class SaveWeightsAfterSteps(Callback):
self.model.save(save_file)
if characters_cnnrnn_ocr:
os.system("cp "+characters_cnnrnn_ocr+" "+os.path.join(os.path.join(self.save_path, f"model_step_{self.step_count}"),"characters_org.txt"))
if self.characters_cnnrnn_ocr:
os.system("cp "+self.characters_cnnrnn_ocr+" "+os.path.join(os.path.join(self.save_path, f"model_step_{self.step_count}"),"characters_org.txt"))
with open(os.path.join(os.path.join(self.save_path, f"model_step_{self.step_count}"),"config_eynollah.json"), "w") as fp:
json.dump(self._config, fp) # encode dict into JSON