Exemplos de load_config em Python

Exemplo n.º 1

def main(opts):

    load_config(opts.config)

    device = torch.device("cuda:0" if (
        torch.cuda.is_available() and config.NUM_GPUS > 0) else "cpu")
    baseline_eval(config, device)

Exemplo n.º 2

Arquivo: calculations.py Projeto: fatihcankurnaz/SensorGAN

def main(opts):

    load_config(opts.config)

    device = torch.device("cuda:0" if (
        torch.cuda.is_available() and config.NUM_GPUS > 0) else "cpu")
    quantitative(config, device)

Exemplo n.º 3

Arquivo: baseline.py Projeto: fatihcankurnaz/SensorGAN

def main(opts):

    load_config(opts.config)

    dataloader = lidar_camera_dataloader(config, transforms_, transforms_seg)
    device = torch.device("cuda:0" if (
        torch.cuda.is_available() and config.NUM_GPUS > 0) else "cpu")
    train(dataloader, config, device)

Exemplo n.º 4

def main(opts):
    load_config(opts.config)
    dataloader = lstm_cycle_dataloader(config)
    device = torch.device("cuda:0" if (
        torch.cuda.is_available() and config.NUM_GPUS > 0) else "cpu")
    train(dataloader, config, device)