Esempi in Python per CppConfig

Esempio n. 1

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)

Esempio n. 2

File: __init__.py Progetto: NoListen/apollocaffe

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)

Esempio n. 3

def set_cpp_loglevel(loglevel):
    CppConfig.set_cpp_loglevel(loglevel)

Esempio n. 4

def set_random_seed(value):
    import numpy as np
    import random
    np.random.seed(value)
    random.seed(value)
    CppConfig.set_random_seed(value)

Esempio n. 5

File: __init__.py Progetto: NoListen/apollocaffe

def set_cpp_loglevel(loglevel):
    CppConfig.set_cpp_loglevel(loglevel)

Esempio n. 6

File: __init__.py Progetto: NoListen/apollocaffe

def set_random_seed(value):
    import numpy as np
    import random
    np.random.seed(value)
    random.seed(value)
    CppConfig.set_random_seed(value)