1. 报错信息
ValueError: (InvalidArgument) The type of data we are trying to retrieve does not match
the type of data currently contained in the container.
[Hint: Expected dtype() == paddle::experimental::CppTypeToDataType<T>::Type(),
but received dtype():11 != paddle::experimental::CppTypeToDataType<T>::Type():10.]
(at ..\paddle\phi\core\dense_tensor.cc:143)
2. 解决办法
这个一般是自定义的数据类型与model
接收的类型不一致,训练时的数据类型是float32
。所以将自定义的DataSet
中返回的numpy
数组的类型转为float32
类型就可以了。