diff options
author | ZJaume <jzaragoza@prompsit.com> | 2021-11-10 17:18:17 +0300 |
---|---|---|
committer | ZJaume <jzaragoza@prompsit.com> | 2021-11-10 17:18:17 +0300 |
commit | 010b11ca1fccae2a8bd6fe271948427057e8ef57 (patch) | |
tree | a7dc864e8a8ec7ed6b7f84e491abfd1a8dd9b26b /bicleaner_ai | |
parent | 5a745dd5811c1b48e47f888d9f3f00320216a561 (diff) |
Avoid write metadata file too early
Diffstat (limited to 'bicleaner_ai')
-rwxr-xr-x | bicleaner_ai/bicleaner_ai_train.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/bicleaner_ai/bicleaner_ai_train.py b/bicleaner_ai/bicleaner_ai_train.py index 5123e46..459811d 100755 --- a/bicleaner_ai/bicleaner_ai_train.py +++ b/bicleaner_ai/bicleaner_ai_train.py @@ -132,8 +132,6 @@ def initialization(): if args.porn_removal_file and args.porn_removal_file.count('/') == 0: args.porn_removal_file = args.model_dir + '/' + args.porn_removal_file - args.metadata = open(args.model_dir + '/metadata.yaml', 'w+') - # Logging logging_setup(args) logging_level = logging.getLogger().level @@ -228,6 +226,7 @@ def perform_training(args): os.unlink(valid_sentences) logging.info("End training") + args.metadata = open(args.model_dir + '/metadata.yaml', 'w+') write_metadata(args, classifier, y_true, y_pred, lm_stats) args.metadata.close() |