Surya Dheeshjith

I am slightly confused as to how the heading angle is calculated here. ``` # Heading if self.rotate_to_center: heading_angle = self.heading_list[index] - rot_angle else: heading_angle = self.heading_list[index] ``` self.heading_list[index] is...

Was a bit confused while testing as the performance was really bad until it hit me that actions are being picked randomly! This should help others I hope.

I have a few queries regarding the CausalSelfAttention module in the codebase. 1. In lines https://github.com/songweige/TATS/blob/main/tats/modules/gpt.py#L100C13-L102C86, ``` mask[:, :config.n_unmasked+1] = 1 mask[:, -config.n_unmasked+1:] = 1 mask[-config.n_unmasked+1:, config.n_unmasked+1:-config.n_unmasked+1] = 0 ```...

Updating comment here due to shape error

Some files need to be modified to take in any kind of agent index. Not just values from 0-n-1.