Python load_config Beispiele

Beispiel #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)

Beispiel #2

Datei: calculations.py Projekt: 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)

Beispiel #3

Datei: baseline.py Projekt: 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)

Beispiel #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)