Argument d_headis not used in TSiTPlus

[vrodriguezf]

tsai/tsai/models/TSiTPlus.py

Line 63 in 6baf0ba

def __init__(self, c_in:int, seq_len:int, depth:int=6, d_model:int=128, n_heads:int=16, d_head:Optional[int]=None, act:str='gelu',

[oguiza]

Good catch, @vrodriguezf!
I've checked the ViT implementation that is the one I used as a blueprint to create TSiT and it doesn't use the d_head argument. So based on this, I decided to drop it. I've already fixed this issue in GitHub.

[oguiza]

I'll close this issue since I've checked that d_head is no longer an argument in TSiT in GitHub.