INNER CODE UNIT · Python

main

dailenson/SDT · user_generate.py:12

def main(opt):
    """ load config file into cfg"""
    cfg_from_file(opt.cfg_file)
    assert_and_infer_cfg()

    """setup data_loader instances"""
    test_dataset = UserDataset(
       cfg.DATA_LOADER.PATH, cfg.DATA_LOADER.DATASET, opt.style_path)
    test_loader = torch.utils.data.DataLoader(test_dataset,
                                              batch_size=cfg.TRAIN.IMS_PER_BATCH,
                                              shuffle=True,
                                              sampler=None,
                                              drop_last=False,
                                              num_workers=cfg.DATA_LOADER.NUM_THREADS)

    os.makedirs(os.path.join(opt.save_dir), exist_ok=True)

    """build model architecture"""

View source record →