DirtyHarryLYL / HAKE-Action-Torch

HAKE-Action in PyTorch

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

evaluate error

buaaswf opened this issue · comments

(hake) dell@dell-07:~/HAKE-Action-Torch-IDN-Integrating-Decomposing-Network-$ sh test.sh
odict_keys(['AE.spatial.fc1.weight', 'AE.spatial.fc1.bias', 'AE.spatial.bn1.weight', 'AE.spatial.bn1.bias', 'AE.spatial.bn1.running_mean', 'AE.spatial.bn1.running_var', 'AE.spatial.bn1.num_batches_tracked', 'AE.spatial.fc2.weight', 'AE.spatial.fc2.bias', 'AE.spatial.bn2.weight', 'AE.spatial.bn2.bias', 'AE.spatial.bn2.running_mean', 'AE.spatial.bn2.running_var', 'AE.spatial.bn2.num_batches_tracked', 'AE.encoder.fc1.weight', 'AE.encoder.fc1.bias', 'AE.encoder.bn1.weight', 'AE.encoder.bn1.bias', 'AE.encoder.bn1.running_mean', 'AE.encoder.bn1.running_var', 'AE.encoder.bn1.num_batches_tracked', 'AE.encoder.fc2.weight', 'AE.encoder.fc2.bias', 'AE.decoder.fc1.weight', 'AE.decoder.fc1.bias', 'AE.decoder.bn1.weight', 'AE.decoder.bn1.bias', 'AE.decoder.bn1.running_mean', 'AE.decoder.bn1.running_var', 'AE.decoder.bn1.num_batches_tracked', 'AE.decoder.fc2.weight', 'AE.decoder.fc2.bias', 'AE.classifier.weight', 'AE.classifier.bias', 'AE.classification_loss.pos_weight', 'diff.trans.trans1.weight', 'diff.trans.trans1.bias', 'diff.trans.bn1.weight', 'diff.trans.bn1.bias', 'diff.trans.bn1.running_mean', 'diff.trans.bn1.running_var', 'diff.trans.bn1.num_batches_tracked', 'diff.trans.trans2.weight', 'diff.trans.trans2.bias', 'classification_loss.pos_weight', 'reverse.trans.trans1.weight', 'reverse.trans.trans1.bias', 'reverse.trans.bn1.weight', 'reverse.trans.bn1.bias', 'reverse.trans.bn1.running_mean', 'reverse.trans.bn1.running_var', 'reverse.trans.bn1.num_batches_tracked', 'reverse.trans.trans2.weight', 'reverse.trans.trans2.bias', 'AE.binary_classifier.weight', 'AE.binary_classifier.bias'])
Traceback (most recent call last):
File "eval.py", line 68, in
net = models[config.MODE](config.MODEL, HO_weight)
File "/home/dell/HAKE-Action-Torch-IDN-Integrating-Decomposing-Network-/model.py", line 245, in init
self.load_state_dict(state)
File "/home/dell/miniconda3/envs/hake/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1045, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for IDN:
Missing key(s) in state_dict: "AE.semihard_loss.pos_weight", "semihard_loss.pos_weight".

Duplicate with #10.