INNER CODE UNIT · Python

iters

pprp/SimpleCVReproduction · NAS/AngleNAS/SPOS/shrinking/main.py:92

    iters = int((config.layers * (config.op_num+1) -
                 len(config.stage_last_id) - 1) / config.per_stage_drop_num)
    args.total_iters = args.epochs * per_epoch_iters
    optimizer, scheduler = get_optimizer_schedule(model, args)

    # Prepare data
    train_loader = get_train_dataloader(
        args.train_dir, args.batch_size, args.local_rank, args.total_iters)
    train_dataprovider = DataIterator(train_loader)

    operations = [list(range(config.op_num)) for i in range(config.layers)]
    for i in range(len(operations)):
        if i not in config.stage_last_id and not i == 0:
            operations[i].append(-1)
    print('operations={}'.format(operations))

    seed = args.seed
    start_iter, ops_dim, modify_initial_model = 0, 0, False

View source record →