From a2eb410cdb7c8c39736ec798c276308d7893581f Mon Sep 17 00:00:00 2001 From: ZJaume Date: Thu, 15 Sep 2022 14:16:39 +0000 Subject: Set SentencePiece seed --- bicleaner_ai/bicleaner_ai_train.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/bicleaner_ai/bicleaner_ai_train.py b/bicleaner_ai/bicleaner_ai_train.py index 32e5018..a9acc4a 100755 --- a/bicleaner_ai/bicleaner_ai_train.py +++ b/bicleaner_ai/bicleaner_ai_train.py @@ -12,6 +12,7 @@ if 'BICLEANER_AI_THREADS' in os.environ: from tempfile import TemporaryFile, NamedTemporaryFile, gettempdir from multiprocessing import cpu_count from timeit import default_timer +import sentencepiece as spm import tensorflow as tf import numpy as np import argparse @@ -112,6 +113,7 @@ def initialization(): random.seed(args.seed) os.environ["PYTHONHASHSEED"] = str(args.seed) tf.random.seed = args.seed + spm.set_random_generator_seed(args.seed) if args.gpu is not None: os.environ["CUDA_VISIBLE_DEVICES"] = str(args.gpu) -- cgit v1.2.3