Function exp_flex_component

scripts/config_generator.py:250–310

(label_amount)

def exp_flex_component(label_amount):
    config_file = r'./config/'
    save_path = r'./saved_models/'

    if not os.path.exists(config_file):
        os.mkdir(config_file)
    if not os.path.exists(save_path):
        os.mkdir(save_path)

    algs = ['uda', 'pseudolabel']
    datasets = ['cifar10', 'cifar100', 'svhn', 'stl10']
    # seeds = [1, 11, 111]
    seeds = [0]

    dist_port = range(11121, 12120, 1)
    count = 0

    for alg in algs:
        for dataset in datasets:
            for seed in seeds:
                # change the configuration of each dataset
                if dataset == 'cifar10':
                    net = 'WideResNet'
                    num_classes = 10
                    num_labels = label_amount[0]
                    weight_decay = 5e-4
                    depth = 28
                    widen_factor = 2
                elif dataset == 'cifar100':
                    net = 'WideResNet'
                    num_classes = 100
                    num_labels = label_amount[1]
                    weight_decay = 1e-3
                    depth = 28
                    widen_factor = 8
                elif dataset == 'svhn':
                    net = 'WideResNet'
                    num_classes = 10
                    num_labels = label_amount[2]
                    weight_decay = 5e-4
                    depth = 28
                    widen_factor = 2
                elif dataset == 'stl10':
                    net = 'WideResNetVar'
                    num_classes = 10
                    num_labels = label_amount[3]
                    weight_decay = 5e-4
                    depth = 28
                    widen_factor = 2

                port = dist_port[count]
                # prepare the configuration file
                cfg = create_base_config(alg, seed,
                                         dataset, net, num_classes, num_labels,
                                         port,
                                         weight_decay, depth, widen_factor
                                         )
                count += 1