I am trying a recurrent network in Linux server using MMA 11.3. The network structure is as follows:
net = NetInitialize[
NetGraph[
<|
"Conv" ->
NetMapOperator[
NetChain[{ConvolutionLayer[3, {3, 3}], FlattenLayer[]}, "Input" -> {5, 11, 11}]],
"LSTM1" -> LongShortTermMemoryLayer[40, "Dropout" -> 0.4],
"Dropout" -> DropoutLayer[0.3],
"Linear" -> NetMapOperator[LinearLayer[1]], "Ramp" -> Ramp,
"Sum" -> SummationLayer[]
|>,
{NetPort["Circulation"] -> "Conv" -> "LSTM1" -> "Dropout" -> "Linear" -> "Ramp"
-> "Sum" -> NetPort["TotalPrecipitation"],
"Ramp" -> NetPort["HourlyPrecipitation"]}
]
]
It works for MMA 11.3 Macbook Pro. However, when I copied the code to the Linux server for training using GPU, it reports that:
NetGraph::valfail: Validation failed for SummationLayer: Layer does not support varying dimensions.
Could anyone help?
Lookup[PacletInformation["NeuralNetworks"], "Version"]
on both machines. Just runPacletUpdate["NeuralNetworks"]
on your Linux machine if older version. $\endgroup$ – Sebastian Jun 22 '18 at 18:19