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

更新文档

Browse Source
This commit is contained in:
sml2h3
2022-02-26 15:09:54 +08:00
parent aa3cdfe9a3
commit 44ac3db79f
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/train.py
+5 -2
View File
@@ -30,6 +30,9 @@ class Train:
self.target_acc = self.target['Accuracy']
self.min_epoch = self.target['Epoch']
self.max_loss = self.target['Cost']
self.resize = [int(self.conf['Model']['ImageWidth']), int(self.conf['Model']['ImageHeight'])]
logger.info("\nTaget:\nmin_Accuracy: {}\nmin_Epoch: {}\nmax_Loss: {}".format(self.target_acc, self.min_epoch,
self.max_loss))
@@ -143,8 +146,8 @@ class Train:
self.project_name, str(accuracy), self.epoch, self.step,
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, "charset.json"), 'w', encoding="utf-8") as f:
f.write(json.dumps(self.net.charset, ensure_ascii=False))
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]}, ensure_ascii=False))
logger.info("\nExport Finished!Using Time: {}min".format(
str(int(int(self.now_time) - int(self.start_time)) / 60)))
exit()