Python serialize_config Examples

Example #1

def main(argv):
    parser = get_argument_parser()
    config = get_config_from_argv(argv, parser)

    serialize_config(config, config.log_dir)

    nncf_root = Path(__file__).absolute().parents[3]
    create_code_snapshot(nncf_root, os.path.join(config.log_dir, "snapshot.tar.gz"))

    run_train(config)

Example #2

File: main.py Project: yiweichen04/nncf

def main(argv):
    parser = get_argument_parser()
    config = get_config_from_argv(argv, parser)

    serialize_config(config, config.log_dir)

    nncf_root = Path(__file__).absolute().parents[3]
    create_code_snapshot(nncf_root, osp.join(config.log_dir,
                                             'snapshot.tar.gz'))
    if 'train' in config.mode or 'test' in config.mode:
        run(config)
    elif 'export' in config.mode:
        export(config)

Example #3

def main(argv):
    parser = get_argument_parser()
    config = get_config_from_argv(argv, parser)

    serialize_config(config, config.log_dir)

    nncf_root = Path(__file__).absolute().parents[3]
    create_code_snapshot(nncf_root,
                         os.path.join(config.log_dir, "snapshot.tar.gz"))

    if config.dataset_type != 'tfrecords':
        raise RuntimeError(
            'The train.py does not support TensorFlow Datasets (TFDS). '
            'Please use TFRecords.')

    run_train(config)