I met a mistake.

[jhzmike]

Traceback (most recent call last):
File "D:\skrsuper\python1\pytorch\cost\train.py", line 97, in
loss_log = model.fit(
File "D:\skrsuper\python1\pytorch\cost\cost.py", line 299, in fit
loss = self.cost(x_q, x_k)
File "C:\Users\lx.conda\envs\py39\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "D:\skrsuper\python1\pytorch\cost\cost.py", line 145, in forward
q_t, q_s = self.encoder_q(x_q)
File "C:\Users\lx.conda\envs\py39\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "D:\skrsuper\python1\pytorch\cost\models\encoder.py", line 163, in forward
out = mod(x) # b t d
File "C:\Users\lx.conda\envs\py39\lib\site-packages\torch\nn\modules\module.py", line 1190, in _call_impl
return forward_call(*input, **kwargs)
File "D:\skrsuper\python1\pytorch\cost\models\encoder.py", line 65, in forward
output_fft[:, self.start:self.end] = self._forward(input_fft)
File "D:\skrsuper\python1\pytorch\cost\models\encoder.py", line 69, in _forward
output = torch.einsum('bti,tio->bto', input[:, self.start:self.end], self.weight)
File "C:\Users\lx.conda\envs\py39\lib\site-packages\torch\functional.py", line 378, in einsum
return _VF.einsum(equation, operands) # type: ignore[attr-defined]
RuntimeError: einsum(): subscript t has size 501 for operand 1 which does not broadcast with previously seen size 301