senta-bilstm在进行迁移训练时报错
Created by: BledCao
您好,请教下,我本来是想按自己的数据集迁移训练模型,但是运行以下代码后总是报错,是因为hub的安装路径和Anaconda的安装路径不同导致的吗?求解~
paddlehub1.7.1 系统:windows python 3.6.1 运行代码: import paddlehub as hub module = hub.Module(name="senta_bilstm") inputs, outputs, program = module.context(trainable=True) from paddlehub.dataset.base_nlp_dataset import BaseNLPDataset
class DemoDataset(BaseNLPDataset): def init(self): # 数据集存放位置 self.dataset_dir = "C:/Users/ccb/data/" super(DemoDataset, self).init( base_path=self.dataset_dir, train_file="train.tsv", dev_file="dev.tsv", test_file="test.tsv", # 如果还有预测数据(不需要文本类别label),可以放在predict.tsv #predict_file="predict.tsv", train_file_with_header=True, dev_file_with_header=True, test_file_with_header=True, #predict_file_with_header=True, # 数据集类别集合 label_list=["0", "1"]) dataset = DemoDataset() reader = hub.reader.LACClassifyReader( dataset=dataset, vocab_path=module.get_vocab_path()) strategy = hub.AdamWeightDecayStrategy( learning_rate=1e-5, weight_decay=0.01, warmup_proportion=0.1, lr_scheduler="linear_decay", )
config = hub.RunConfig(use_cuda=False, num_epoch=3, batch_size=8, strategy=strategy) sent_feature = outputs["sentence_feature"]
feed_list = [inputs["words"].name]
cls_task = hub.TextClassifierTask( data_reader=reader, feature=sent_feature, feed_list=feed_list, num_classes=dataset.num_labels, config=config) run_states = cls_task.finetune_and_eval()
运行后报错信息:
Error Message Summary:
InvalidArgumentError: The Tensor in the scale Op's Input Variable X(@HUB_senta_bilstm@fc_12.w_0) is not initialized. [Hint: Expected t->IsInitialized() == true, but received t->IsInitialized():0 != true:1.] at (D:\1.8.1\paddle\paddle\fluid\framework\operator.cc:1289) [operator < scale > error]