INNER CODE UNIT · Python

scheduler

pprp/SimpleCVReproduction · NAS/AngleNAS/FairNAS/searching/main.py:96

    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)

    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()))
        save_checkpoint({
        'topk':topk,
        'state_dict': model.state_dict(),

View source record →