yan/dddd_trainer
1
0
Fork 0
mirror of https://github.com/netfun2000/dddd_trainer.git synced 2026-08-12 21:09:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

更新文档

Browse Source
This commit is contained in:
sml2h3
2022-02-26 15:15:47 +08:00
parent 0be05695bc
commit 9dedc81912
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/train.py
+2 -1
View File
@@ -33,6 +33,7 @@ class Train:
self.resize = [int(self.conf['Model']['ImageWidth']), int(self.conf['Model']['ImageHeight'])]
self.word = self.conf['Model']['Word']
self.ImageChannel = self.conf['Model']['ImageChannel']
logger.info("\nTaget:\nmin_Accuracy: {}\nmin_Epoch: {}\nmax_Loss: {}".format(self.target_acc, self.min_epoch,
self.max_loss))
@@ -147,7 +148,7 @@ class Train:
time.strftime("%Y-%m-%d-%H-%M-%S", time.localtime(self.now_time))))
, input_names, output_names, dynamic_ax)
with open(os.path.join(self.models_path, "charsets.json"), 'w', encoding="utf-8") as f:
f.write(json.dumps({"charset": self.net.charset, "image": self.resize, "word": self.word}, ensure_ascii=False))
f.write(json.dumps({"charset": self.net.charset, "image": self.resize, "word": self.word, 'channel': self.ImageChannel}, ensure_ascii=False))
logger.info("\nExport Finished!Using Time: {}min".format(
str(int(int(self.now_time) - int(self.start_time)) / 60)))
exit()