Error in training LSTM model #9097
Description
val model = Sequential[Float]()
.add(LSTM(inputSize = 3, hiddenSize = 50))
.add(Linear(inputSize = 50, outputSize = 10))
// .add(LogSoftMax())
val optimizer = Optimizer(model = model,
sampleRDD = data,
criterion = MSECriterion[Float](),
batchSize = 10)
optimizer
.setOptimMethod(new Adam(0.01))
.setEndWhen(Trigger.maxEpoch(10))
.optimize()
data的格式是data: RDD[Sample[Float]]
训练模型报错
java.lang.ClassCastException: com.intel.analytics.bigdl.tensor.DenseTensor cannot be cast to com.intel.analytics.bigdl.utils.Table
at com.intel.analytics.bigdl.nn.Cell.updateOutput(Cell.scala:48)
at com.intel.analytics.bigdl.nn.abstractnn.AbstractModule.forward(AbstractModule.scala:282)
at com.intel.analytics.bigdl.nn.Sequential.updateOutput(Sequential.scala:39)
at com.intel.analytics.bigdl.nn.abstractnn.AbstractModule.forward(AbstractModule.scala:282)
和
23/10/07 18:00:14 ERROR [Executor task launch worker for task 4.0 in stage 14.0 (TID 26)] Executor: Exception in task 4.0 in stage 14.0 (TID 26)
com.intel.analytics.bigdl.utils.LayerException: null
at com.intel.analytics.bigdl.nn.abstractnn.AbstractModule.forward(AbstractModule.scala:288) ~[bigdl-SPARK_3.1-0.13.0.jar:?]
at com.intel.analytics.bigdl.nn.Sequential.updateOutput(Sequential.scala:39) ~[bigdl-SPARK_3.1-0.13.0.jar:?]
at com.intel.analytics.bigdl.nn.abstractnn.AbstractModule.forward(AbstractModule.scala:282) ~[bigdl-SPARK_3.1-0.13.0.jar:?]