exarl.candlelib.pytorch_utils

Module Contents

Functions

set_parallelism_threads()
set_seed(seed)	Set the random number seed to the desired value
get_function(name)
build_activation(type)
build_optimizer(model, type, lr, kerasDefaults, trainable_only=True)
initialize(weights, type, kerasDefaults, seed=None, constant=0.0)
xent(y_true, y_pred)
mse(y_true, y_pred)

exarl.candlelib.pytorch_utils.set_parallelism_threads()

exarl.candlelib.pytorch_utils.set_seed(seed)

Set the random number seed to the desired value

Parameters

seed (integer) – Random number seed.

exarl.candlelib.pytorch_utils.get_function(name)

exarl.candlelib.pytorch_utils.build_activation(type)

exarl.candlelib.pytorch_utils.build_optimizer(model, type, lr, kerasDefaults, trainable_only=True)

exarl.candlelib.pytorch_utils.initialize(weights, type, kerasDefaults, seed=None, constant=0.0)

exarl.candlelib.pytorch_utils.xent(y_true, y_pred)

exarl.candlelib.pytorch_utils.mse(y_true, y_pred)