imagenet_valid.py 文件源码

def run(title, base_batch_size, base_labeled_batch_size, base_lr, data_seed, **kwargs):
    LOG.info('run title: %s', title)
    ngpu = torch.cuda.device_count()
    adapted_args = {
        'batch_size': base_batch_size * ngpu,
        'labeled_batch_size': base_labeled_batch_size * ngpu,
        'lr': base_lr * ngpu,
        'labels': 'data-local/labels/ilsvrc2012/128000_balanced_labels/{:02d}.txt'.format(data_seed),
    }
    context = RunContext(__file__, data_seed)
    main.args = parse_dict_args(**adapted_args, **kwargs)
    main.main(context)