Hello @upeya ,
You forget to do this step.
Our demo model.
Another method. You can modify demo to change the dimensions after getting outputs.
- input0_data = input0_data.reshape(SPAN, LISTSIZE, GRID0, GRID0)
- input1_data = input1_data.reshape(SPAN, LISTSIZE, GRID1, GRID1)
- input2_data = input2_data.reshape(SPAN, LISTSIZE, GRID2, GRID2)
+ input0_data = input0_data.reshape(SPAN, GRID0, GRID0, LISTSIZE)
+ input1_data = input1_data.reshape(SPAN, GRID1, GRID1, LISTSIZE)
+ input2_data = input2_data.reshape(SPAN, GRID2, GRID2, LISTSIZE)
- input_data.append(np.transpose(input0_data, (2, 3, 0, 1)))
- input_data.append(np.transpose(input1_data, (2, 3, 0, 1)))
- input_data.append(np.transpose(input2_data, (2, 3, 0, 1)))
+ input_data.append(np.transpose(input0_data, (1, 2, 0, 3)))
+ input_data.append(np.transpose(input1_data, (1, 2, 0, 3)))
+ input_data.append(np.transpose(input2_data, (1, 2, 0, 3)))
And remember modify LISTSIZE in demo.