INNER CODE UNIT · Python

scheduler

pprp/SimpleCVReproduction · NAS/AngleNAS/SPOS/searching/main.py:94

    scheduler = torch.optim.lr_scheduler.LambdaLR(optimizer, lambda step : (1.0-step/args.total_iters), last_epoch=-1)
    
    # Prepare data
    train_loader = get_train_dataloader(args.train_dir, args.batch_size, args.local_rank, args.total_iters)
    train_dataprovider = DataIterator(train_loader)
    val_loader = get_val_dataloader(args.test_dir)
    val_dataprovider = DataIterator(val_loader)
    
    # load the shrunk search space found by ABS
    operations = pickle.load(open(args.operations_path, 'rb'))
    print('operations={}'.format(operations))
    train(train_dataprovider, val_dataprovider, optimizer, scheduler, model, criterion_smooth, args, val_iters, args.seed, operations)

    if args.local_rank == 0:
        save(model.module, config.net_cache)
        evolution_trainer = EvolutionTrainer()
        topk = evolution_trainer.search(operations)
        now = time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time()))

View source record →