Skip to content

Instantly share code, notes, and snippets.

@gaphex

gaphex/build_bert_module.py

Created February 26, 2020 10:08
Show Gist options

  • Select an option

    Learn more about clone URLs
  • Save gaphex/4c55d12c5907fd9f0f1e01ae734ba3fd to your computer and use it in GitHub Desktop.

Select an option

Learn more about clone URLs
Save gaphex/4c55d12c5907fd9f0f1e01ae734ba3fd to your computer and use it in GitHub Desktop.
Download ZIP
Raw
build_bert_module.py
from loader import load_sts, load_sick2014
from bert_experimental.finetuning.text_preprocessing import build_preprocessor
from bert_experimental.finetuning.bert_layer import BertLayer
from bert_experimental.finetuning.modeling import BertConfig, BertModel, build_bert_module
BERT_DIR = "/content/uncased_L-12_H-768_A-12/"
build_bert_module(BERT_DIR+"bert_config.json",
BERT_DIR+"vocab.txt",
BERT_DIR+"bert_model.ckpt",
"bert_module")
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment