17b34c69d4bc8128ae6d4dba55e209ae34bd8821,examples/basics/svhn-digit-convnet.py,,,#,89

Before Change

    logger.auto_set_dir()
    data_train, data_test = get_data()

    config = TrainConfig(
        model=Model(),
        data=QueueInput(data_train),
        callbacks=[
            ModelSaver(),
            InferenceRunner(data_test,
                            ScalarStats(["cost", "accuracy"]))
        ],
        max_epoch=350,
        session_init=SaverRestore(args.load) if args.load else None
    )
    launch_train_with_config(config, SimpleTrainer())

After Change

    logger.auto_set_dir()
    data_train, data_test = get_data()

    config = TrainConfig(
        model=Model(),
        data=QueueInput(data_train),
        callbacks=[
            ModelSaver(),
            InferenceRunner(data_test,
                            ScalarStats(["cost", "accuracy"]))
        ],
        max_epoch=350,
        session_init=SmartInit(args.load)
    )
    launch_train_with_config(config, SimpleTrainer())

Instances