Python set_parallelism_threads示例

示例#1

文件： uno_baseline_keras2.py 项目： zackcornelius/Benchmarks

def set_seed(seed):
    os.environ['PYTHONHASHSEED'] = '0'
    np.random.seed(seed)

    random.seed(seed)

    if K.backend() == 'tensorflow':
        import tensorflow as tf
        tf.set_random_seed(seed)
        candle.set_parallelism_threads()

示例#2

文件： p1b1.py 项目： zackcornelius/Benchmarks

import numpy as np

from sklearn.metrics import mean_squared_error
from sklearn.metrics import r2_score
from scipy.stats.stats import pearsonr

file_path = os.path.dirname(os.path.realpath(__file__))
#lib_path = os.path.abspath(os.path.join(file_path, '..'))
#sys.path.append(lib_path)
lib_path2 = os.path.abspath(os.path.join(file_path, '..', '..', 'common'))
sys.path.append(lib_path2)

import candle

logger = logging.getLogger(__name__)
candle.set_parallelism_threads()

additional_definitions = [
    {
        'name': 'latent_dim',
        'action': 'store',
        'type': int,
        'help': 'latent dimensions'
    },
    {
        'name': 'model',
        'default': 'ae',
        'choices': ['ae', 'vae', 'cvae'],
        'help': 'model to use: ae, vae, cvae'
    },
    {