Got Error with foot sliding loss

[TongbaoYang]

I got a RuntimeError when running trian.py, it seems that the shepe of pred_vel and contact not matched. Here are the infomation:
我照着程序跑的时候出问题了，应该是计算foot sliding loss中的两个张量形状不一样。报错信息如下：

  File "C:\Users\smki\anaconda3\envs\dl\lib\runpy.py", line 265, in run_path
    return _run_module_code(code, init_globals, run_name,
  File "C:\Users\ytb\smki\envs\dl\lib\runpy.py", line 97, in _run_module_code
    _run_code(code, mod_globals, init_globals,
  File "C:\Users\smki\anaconda3\envs\dl\lib\runpy.py", line 87, in _run_code
    exec(code, run_globals)
  File "d:\UserProfile\Documents\GitDownload\Pytorch-Robust-Motion-In-betweening\train.py", line 300, in <module>
    loss_slide = torch.mean(torch.abs(pred_vel * contact[:,:-1].permute(0, 2, 1).unsqueeze(2)))
RuntimeError: The size of tensor a (9) must match the size of tensor b (49) at non-singleton dimension 3