blob: 774104102ab7559ed9d636862e1faf42738099fe (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
|
#!/usr/bin/env python
import os
# Suppress Tenssorflow logging messages unless log level is explictly set
if 'TF_CPP_MIN_LOG_LEVEL' not in os.environ:
os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
# Set Tensorflow max threads before initialization
if 'BICLEANER_AI_THREADS' in os.environ:
threads = int(os.environ["BICLEANER_AI_THREADS"])
import tensorflow as tf
tf.config.threading.set_intra_op_parallelism_threads(threads)
tf.config.threading.set_inter_op_parallelism_threads(threads)
import sys
import bicleaner_ai.bicleaner_ai_train as train
def main(argv):
args = train.get_arguments()
train.initialization(args)
train.main(args)
if __name__ == '__main__':
main(sys.argv[1:])
|
