예제 #1
0
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)
예제 #2
0
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)
예제 #3
0
def set_cpp_loglevel(loglevel):
    CppConfig.set_cpp_loglevel(loglevel)
예제 #4
0
def set_random_seed(value):
    import numpy as np
    import random
    np.random.seed(value)
    random.seed(value)
    CppConfig.set_random_seed(value)
예제 #5
0
def set_cpp_loglevel(loglevel):
    CppConfig.set_cpp_loglevel(loglevel)
예제 #6
0
def set_random_seed(value):
    import numpy as np
    import random
    np.random.seed(value)
    random.seed(value)
    CppConfig.set_random_seed(value)