faceboxes icon indicating copy to clipboard operation
faceboxes copied to clipboard

I have something wrong when I try to load weight to train the model

Open wnov opened this issue 6 years ago • 1 comments

I get this when I run trainvisdom.py ,shall I do some preparation before running the script??

Traceback (most recent call last): File "E:\My_Project\face_recognition\faceboxes-master\trainvisdom.py", line 112, in train() File "E:\My_Project\face_recognition\faceboxes-master\trainvisdom.py", line 27, in train net.load_state_dict(torch.load('weight/faceboxes.pt')) File "C:\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 721, in load_state_dict self.class.name, "\n\t".join(error_msgs))) RuntimeError: Error(s) in loading state_dict for FaceBox: Unexpected key(s) in state_dict: "bn1.num_batches_tracked", "bn2.num_batches_tracked", "inception1.conv1.1.num_batches_tracked", "inception1.conv2.1.num_batches_tracked", "inception1.conv3.1.num_batches_tracked", "inception1.conv4.1.num_batches_tracked", "inception1.conv5.1.num_batches_tracked", "inception1.conv6.1.num_batches_tracked", "inception1.conv7.1.num_batches_tracked", "inception2.conv1.1.num_batches_tracked", "inception2.conv2.1.num_batches_tracked", "inception2.conv3.1.num_batches_tracked", "inception2.conv4.1.num_batches_tracked", "inception2.conv5.1.num_batches_tracked", "inception2.conv6.1.num_batches_tracked", "inception2.conv7.1.num_batches_tracked", "inception3.conv1.1.num_batches_tracked", "inception3.conv2.1.num_batches_tracked", "inception3.conv3.1.num_batches_tracked", "inception3.conv4.1.num_batches_tracked", "inception3.conv5.1.num_batches_tracked", "inception3.conv6.1.num_batches_tracked", "inception3.conv7.1.num_batches_tracked", "conv3_1.1.num_batches_tracked", "conv3_2.1.num_batches_tracked", "conv4_1.1.num_batches_tracked", "conv4_2.1.num_batches_tracked".

wnov avatar Nov 26 '18 05:11 wnov

Maybe your pytorch version is different from mine.

XiaXuehai avatar Nov 30 '18 03:11 XiaXuehai