Based on the code snippet it seems that the model architecture is equal, but the torch.cat call uses a different order of inputs. Could you explain how dim1 is treated in model.low_level? Would it be the temporal dimension and thus the LSTM would see a different sequence?