Open YuQi9797 opened 3 years ago
作者您好,根据所给源码进行训练后,模型所保存的权重仅在训练阶段进行了保存。
但在测试时,并没有对应的weights: ./pre_train/ki_agcn.pt。 https://github.com/lshiwjx/2s-AGCN/blob/master/config/kinetics-skeleton/test_joint.yaml#L23
weights: ./pre_train/ki_agcn.pt
同时在测试时,模型加载阶段 https://github.com/lshiwjx/2s-AGCN/blob/master/main.py#L249,由于arg.weights也不对应所以出错。请问如何解决呢?
作者您好,根据所给源码进行训练后,模型所保存的权重仅在训练阶段进行了保存。
但在测试时,并没有对应的
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也不对应所以出错。请问如何解决呢?