def set_device(device_id=-1): if device_id == -1: device_string = "CPU device" else: device_string = "GPU device %d" % device_id print("%s - %s" % \ (strftime("%Y-%m-%d %H:%M:%S"), device_string)) CppConfig.set_device(device_id)
def set_device(device_id=-1): if device_id == -1: device_string = "CPU device" else: device_string = "GPU device %d" % device_id print("%s - %s" % \ (strftime("%Y-%m-%d %H:%M:%S"), device_string)) CppConfig.set_device(device_id)
def set_cpp_loglevel(loglevel): CppConfig.set_cpp_loglevel(loglevel)
def set_random_seed(value): import numpy as np import random np.random.seed(value) random.seed(value) CppConfig.set_random_seed(value)
def set_cpp_loglevel(loglevel): CppConfig.set_cpp_loglevel(loglevel)
def set_random_seed(value): import numpy as np import random np.random.seed(value) random.seed(value) CppConfig.set_random_seed(value)