训练问题

[sunqing-code]

对于骨架训练的命令CUDA_VISIBLE_DEVICES=2,3 python main_seg8_v3_pku_pre.py pku joint /data/cq/pku_hrnet_dict/xsub --gd 20 --lr 0.01 --lr_steps 25 30 35 --epochs 40 -b 64 --seg 8 --dropout 0.9 --model_name pku_xsub_sparse_msst_v3_seg8中，model “pku_xsub_sparse_msst_v3_seg8”是如何获得的呢？对于ntu数据集，应该使用什么模型呢？

[Maxchengqin]

pku_xsub_sparse_msst_v3_seg8需要训练出来，命令行里model_name的意思是模型文件(.pth.tar形式)的保存名字。训练使用的模型都是一样的，msst_seg8_v3_pre_replace_fc.py, 在ops文件夹里。