Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

SpatioTemporal Transformer Model #75

Merged
merged 2 commits into from
Jan 31, 2024
Merged

SpatioTemporal Transformer Model #75

merged 2 commits into from
Jan 31, 2024

Conversation

kaseris
Copy link
Owner

@kaseris kaseris commented Jan 31, 2024

Now that a new player has joined in the training of skeleton forecasting, Human3.6M, I had to adapt the model for predicting different sized vectors. In the case of Human3.6M, the predicted vectors should be 4D. I also had to cater for the case of SMPL models, where I do not exactly know how their skeleton is parameterized. So adding input_dim as a parameter to both the model and in the config files should do the job.

Small PR, but will do the job.

@kaseris kaseris merged commit c711edb into main Jan 31, 2024
@kaseris kaseris deleted the fix/sttf branch January 31, 2024 09:53
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant