Why train.py always gets stuck on pretraining with MSE loss

nachiket92 / conv-social-pooling

Code for model proposed in: Nachiket Deo and Mohan M. Trivedi,"Convolutional Social Pooling for Vehicle Trajectory Prediction." CVPRW, 2018

MIT License

336 stars 102 forks source link

Open ruicongzhang opened 3 months ago

ruicongzhang commented 3 months ago

I already have .mat files, but when I run train.py, it always gets stuck. I'm not sure if I'm missing some steps or operations