训练完成后，执行 testXX.yaml 出错。

[YuQi9797]

作者您好，根据所给源码进行训练后，模型所保存的权重仅在训练阶段进行了保存。

但在测试时，并没有对应的weights: ./pre_train/ki_agcn.pt。 https://github.com/lshiwjx/2s-AGCN/blob/master/config/kinetics-skeleton/test_joint.yaml#L23

同时在测试时，模型加载阶段 https://github.com/lshiwjx/2s-AGCN/blob/master/main.py#L249，由于arg.weights也不对应所以出错。请问如何解决呢？