test.py _ classifier.weight

[BoostMom]

hello,

  very nice work~
  I am running the demo with CUB data, however, in test stage, an error happend:(And I found that the code just updated a day ago~)

Traceback (most recent call last):
File "test.py", line 108, in
model.load_state_dict(tmp['state'])
File "/root/anaconda3/lib/python3.6/site-packages/torch/nn/modules/module.py", line 769, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for BaselineFinetune:
Unexpected key(s) in state_dict: "classifier.weight", "classifier.bias".

[wyharveychen]

Hello, I have corrected it. Please use the new test.py, thanks!