M2I is a simple but effective joint motion prediction framework through marginal and conditional predictions by exploiting the factorized relations between interacting agents.
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?
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?