在SKEP-ERNIE1.0中文模型的基础上,继续pre-train
这一步报错
Traceback (most recent call last):
File "pretraining.py", line 359, in
main(args)
File "pretraining.py", line 351, in main
trainer = trainer_class(params, readers, model)
File "pretraining.py", line 152, in init
BaseTrainer.init(self, params, data_set_reader, model_class)
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 48, in init
self.init_net()
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 137, in init_net
self.init_train_net()
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 161, in init_train_net
*opt_args)
File "/home/pengjm/Senta/senta/training/basetrainer.py", line 776, in optimization
, param_grads = optimizer.minimize(loss)
File "", line 2, in minimize
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 277, in impl
return func(args, **kwargs)
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/optimizer.py", line 835, in minimize
no_grad_set=no_grad_set)
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/optimizer.py", line 673, in backward
loss.shape)
AssertionError: The loss.shape should be (1L,), but the current loss.shape is (-1,). Maybe that you should call fluid.layers.mean to process the current loss.
请问是什么原因?
在SKEP-ERNIE1.0中文模型的基础上,继续pre-train 这一步报错 Traceback (most recent call last): File "pretraining.py", line 359, in
main(args)
File "pretraining.py", line 351, in main
trainer = trainer_class(params, readers, model)
File "pretraining.py", line 152, in init
BaseTrainer.init(self, params, data_set_reader, model_class)
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 48, in init
self.init_net()
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 137, in init_net
self.init_train_net()
File "/home/pengjm/Senta/senta/training/base_trainer.py", line 161, in init_train_net
*opt_args)
File "/home/pengjm/Senta/senta/training/basetrainer.py", line 776, in optimization
, param_grads = optimizer.minimize(loss)
File "", line 2, in minimize
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/dygraph/base.py", line 277, in impl
return func( args, **kwargs)
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/optimizer.py", line 835, in minimize
no_grad_set=no_grad_set)
File "/home/pengjm/.virtualenvs/paddle-venv/lib/python3.7/site-packages/paddle/fluid/optimizer.py", line 673, in backward
loss.shape)
AssertionError: The loss.shape should be (1L,), but the current loss.shape is (-1,). Maybe that you should call fluid.layers.mean to process the current loss.
请问是什么原因?