INNER CODE UNIT · Python

per_stage_iters

pprp/SimpleCVReproduction · NAS/AngleNAS/DARTS/shrinking/main.py:120

        per_stage_iters = config.other_stage_epochs * per_epoch_iters if i > 0 else config.first_stage_epochs * per_epoch_iters
        seed = train(train_dataprovider, optimizer, scheduler, model, criterion_smooth, operations, i, per_stage_iters, seed, args)

        if args.local_rank == 0:
            # Search space shrinking
            load(base_model, config.base_net_cache)
            operations = ABS(base_model, model.module, operations, i)
            now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))
            print('{} |=> iter = {}, operations={}, seed={}'.format(now, i+1, operations, seed))

            save_checkpoint( { 'operations':operations,
                                'iter':i,
                                'state_dict': model.state_dict(),
                                'seed':seed
                             }, config.checkpoint_cache)

            operations = merge_ops(operations)
            ops_dim = len(operations)

View source record →