bidireaction-trajectory-prediction icon indicating copy to clipboard operation
bidireaction-trajectory-prediction copied to clipboard

Published 20 hours ago • verified publisher icon umautobots

Trouble Training BiTrapGMM on JAAD and PIE

Open werdnabae opened this issue 2 years ago • 0 comments

Hello authors, I'm trying to train the BiTrapGMM on JAAD and PIE, but I'm running into some troubles while doing so. This is the error message I'm getting. image

Changing line 32 in bitrap_gmm.py to: self.node_future_encoder_h = nn.Linear(self.cfg.GLOBAL_INPUT_DIM, 32) solves this issue, but raises a new issue later in the forward method of bitrap_gmm.py image

Any insight into these issues would be appreciated!

Thank you, Andrew

werdnabae avatar Jul 14 '22 16:07 werdnabae