Bert_model_cfg = './bert_base.json'
vocab = './vocab.txt'
DexBERT_file = './model_steps_604364.pt'
# model initialization
batch_size = 32
Bert_model_cfg = Config.from_json(Bert_model_cfg)
BertAE = DexBERT(Bert_model_cfg)
print(f"Loading model on {device}...")
BertAE.load_state_dict(torch.load(DexBERT_file, map_location=device), strict=False)
BertAE.to(device)
BertAE.eval()
Unfortunately, the required model file model_steps_604364.pt is not provided, rendering the entire model inference process non-functional.
Bert_model_cfg = './bert_base.json'
vocab = './vocab.txt'
DexBERT_file = './model_steps_604364.pt'
Unfortunately, the required model file
model_steps_604364.ptis not provided, rendering the entire model inference process non-functional.