Confusion about the swin transformer code.

Petrichor625 / HLTP

[IEEE TIV] Official PyTorch Implementation of ''A Cognitive-Based Trajectory Prediction Approach for Autonomous Driving.''

32 stars 3 forks source link

Confusion about the swin transformer code. #5

Open guoyage opened 4 months ago

guoyage commented 4 months ago

Hello author, thank you for your wonderful research. I have a small problem. As can be seen from the research results of your paper, swin transformer module has a great impact on the performance of your model. When I try to adopt swin transformer module, I find that your code is missing the relevant codes of self.weights. I used a simple linear layer instead, but the results are not ideal. I hope you can clarify the source code of your self.weights, looking forward to your reply.

yachour1 commented 1 week ago

Hey, did you find a solution, try to reach me on linkedin https://www.linkedin.com/in/yassine-achour2710/