search

abduallahmohamed / Social-STGCNN

Code for "Social-STGCNN: A Social Spatio-Temporal Graph Convolutional Neural Network for Human Trajectory Prediction" CVPR 2020
MIT License
483 stars 141 forks source link

About num_epochs? #53

Closed daeunni closed 2 years ago

daeunni commented 2 years ago

Hello, thank you for your great work. : )

Why did you set the epoch size to 250? When I run it on my computer, I check that some overfitting has occurred, so I want to reduce this epoch size. I wonder if there is a problem with applying the Social STGCNN algorithm even if I reduce it.

abduallahmohamed commented 2 years ago

Thanks for reaching out. I don't think there's an issue if you reduced the epoch size to counter overfitting.

abduallahmohamed commented 2 years ago

I'm also wondering where did you detect overfitting?

daeunni commented 2 years ago

Thank you for your answer. :)

I'm currently making another model based on Social STGCNN, so I just wanted to know if there was any technical problem with Social STGCNN based on my model when I reduced the epoch.

Therefore, overfitting may not be a problem of Social STGCNN. Thanks.