ViTPose icon indicating copy to clipboard operation
ViTPose copied to clipboard

Apex is not installed when running the demo

Open jun297 opened this issue 1 year ago • 2 comments

Hi, when I run demo under slurm environment with the python command, this will be ran via sbatch command:

python demo/top_down_video_demo_with_mmdet.py \
	mmdet_configs/yolox/yolox_tiny_8x8_300e_coco.py \
	pretrained_model/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth \
	configs/body/2d_kpt_sview_rgb_img/topdown_heatmap/coco/ViTPose_base_coco_256x192.py \
	pretrained_model/vitpose_base_coco_aic_mpii.pth \
	--video-path https://user-images.githubusercontent.com/87690686/137440639-fb08603d-9a35-474e-b65f-46b5c06b68d6.mp4 \
	--out-video-root vis_results

Then, I get a warning:

apex is not installed
apex is not installed
apex is not installed
load checkpoint from local path: pretrained_model/yolox_tiny_8x8_300e_coco_20211124_171234-b4047906.pth
load checkpoint from local path: pretrained_model/vitpose_base_coco_aic_mpii.pth
The model and loaded state dict do not match exactly

unexpected key in source state_dict: associate_keypoint_heads.0.deconv_layers.0.weight, associate_keypoint_heads.0.deconv_layers.1.weight, associate_keypoint_heads.0.deconv_layers.1.bias, associate_keypoint_heads.0.deconv_layers.1.running_mean, associate_keypoint_heads.0.deconv_layers.1.running_var, associate_keypoint_heads.0.deconv_layers.1.num_batches_tracked, associate_keypoint_heads.0.deconv_layers.3.weight, associate_keypoint_heads.0.deconv_layers.4.weight, associate_keypoint_heads.0.deconv_layers.4.bias, associate_keypoint_heads.0.deconv_layers.4.running_mean, associate_keypoint_heads.0.deconv_layers.4.running_var, associate_keypoint_heads.0.deconv_layers.4.num_batches_tracked, associate_keypoint_heads.0.final_layer.weight, associate_keypoint_heads.0.final_layer.bias, associate_keypoint_heads.1.deconv_layers.0.weight, associate_keypoint_heads.1.deconv_layers.1.weight, associate_keypoint_heads.1.deconv_layers.1.bias, associate_keypoint_heads.1.deconv_layers.1.running_mean, associate_keypoint_heads.1.deconv_layers.1.running_var, associate_keypoint_heads.1.deconv_layers.1.num_batches_tracked, associate_keypoint_heads.1.deconv_layers.3.weight, associate_keypoint_heads.1.deconv_layers.4.weight, associate_keypoint_heads.1.deconv_layers.4.bias, associate_keypoint_heads.1.deconv_layers.4.running_mean, associate_keypoint_heads.1.deconv_layers.4.running_var, associate_keypoint_heads.1.deconv_layers.4.num_batches_tracked, associate_keypoint_heads.1.final_layer.weight, associate_keypoint_heads.1.final_layer.bias, backbone.cls_token

Although I guess I can get the right result, why does this warning show?

jun297 avatar May 16 '23 10:05 jun297

@jun297 Sorry to bother,have you addressed the problem?

Polaris7777777 avatar Jul 20 '23 06:07 Polaris7777777

@Polaris7777777 No, I have not.

jun297 avatar Jul 20 '23 10:07 jun297