We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
python test.py --cfg configs/mobilenet_v2_CNFx3_coco_384x288.yaml --exp_name o/mobile_cnf3_coco/ --test_model weights/pnfs-mobilenet-v2_nfx3_384x288_coco_ckpt.tar --flip_test the error is: RuntimeError: Error(s) in loading state_dict for Body_Part_Representation: Missing key(s) in state_dict: "total_ops", "total_params", "backbone.total_ops", "backbone.total_params", "backbone.features.total_ops", "backbone.features.total_para ms", "backbone.features.1.total_ops", "backbone.features.1.total_params".......
The mobilenet_v2_CNFx3_coco_384x288.yaml file is: model: use_pretrained: false pretrained: '' use_backbone: True frozen_resnet: False frozen_mobilenet: False backbone_net_name: 'mobilenet_v2' # or 'meta_arch' or 'mobilenet_v2' #backbone_layers: 50 #resnet backbone_feature_num: 4 # resnet backbone_pretrained_path: './weights/mobilenet_v2.pth.tar' init_weights: True dataset_name: 'coco' keypoints_num: 17 input_size: w: 288 h: 384 heatmap_size: w: 72 h: 96 margin_to_border: 1.25 # >=1
#Only when using an extra mask module branch, NOT USED IN THE PAPER use_mask: false mask_num: 0 backbone: frozen_backbone: false # search_alpha: True search_beta: True operators: ["Zero", "skip_connect", "Sep_Conv_3x3", "Atr_Conv_3x3","avg_pool_3x3","max_pool_3x3"] reserve_layers_num: 4 # only use first 4 layers depth: 12 size_types: [4,8,16,32] hidden_states_num: 1 factor: 8 input_nodes_num: 1 subnetwork_config: dataset_name: 'coco' parts_num : 3 cell_config: vector_in_pixel : True vector_dim: 8 convolution_mode: '2D' one-shot-search: True search_alpha: true search_beta: true operators: [ "skip_connect", "Sep_Conv_3x3", "Atr_Conv_3x3",] depth: 6 cut_layers_num: 3 # first several layers size_types: [4,8,16,32] hidden_states_num: 1 factor: 16 input_nodes_num: 1
Thanks!
The text was updated successfully, but these errors were encountered:
No branches or pull requests
python test.py --cfg configs/mobilenet_v2_CNFx3_coco_384x288.yaml --exp_name o/mobile_cnf3_coco/ --test_model weights/pnfs-mobilenet-v2_nfx3_384x288_coco_ckpt.tar --flip_test
the error is:
RuntimeError: Error(s) in loading state_dict for Body_Part_Representation:
Missing key(s) in state_dict: "total_ops", "total_params", "backbone.total_ops", "backbone.total_params", "backbone.features.total_ops", "backbone.features.total_para
ms", "backbone.features.1.total_ops", "backbone.features.1.total_params".......
The mobilenet_v2_CNFx3_coco_384x288.yaml file is:
model:
use_pretrained: false
pretrained: ''
use_backbone: True
frozen_resnet: False
frozen_mobilenet: False
backbone_net_name: 'mobilenet_v2' # or 'meta_arch' or 'mobilenet_v2'
#backbone_layers: 50 #resnet
backbone_feature_num: 4 # resnet
backbone_pretrained_path: './weights/mobilenet_v2.pth.tar'
init_weights: True
dataset_name: 'coco'
keypoints_num: 17
input_size:
w: 288
h: 384
heatmap_size:
w: 72
h: 96
margin_to_border: 1.25 # >=1
Thanks!
The text was updated successfully, but these errors were encountered: