import tensorflow as tf

        params.output,
        "%s.json" % args.model,
        collect_params(params, model_cls.get_parameters())
    )

    # Build Graph
    with tf.Graph().as_default():
        if not params.record:
            # Build input queue
            if params.use_bert and params.bert_emb_path:
                features = dataset.get_training_input_with_bert(params.input + [params.bert_emb_path], params)
            else:
