Changing hidden size leads to runtime error

[Yang-Li-2000]

When training the conditional predictor, changing the hidden_size from 128 to 64 results in an runtime error: RuntimeError: The expanded size of the tensor (64) must match the existing size (128) at non-singleton dimension 1. Target sizes: [10, 64]. Tensor sizes: [10, 128]

Similarly, when training the relation prediction, hidden size == 64 does not work.

What other hyper parameters do I need to change simultaneously when changing hidden_size?