Parameter size of the loaded URL classifier doesn't match with that of model

[RongKaiWeskerMA]

Hi, thanks for open-source the code!

I am trying to understand the implementation of the work. However, when I load the parameters of f_{\phi} from the checkpoint, the classifier size doesn't match with the size of the classifier of the model.

eg, the size of the loaded classifier is (num_c1+num_c2+...+num_c8----->512) but the model has 8 classifier, eg, (num_c1----->512) (num_c2----->512) ....(num_c8----->512)

[RongKaiWeskerMA]

Sorry, it is due to my own operation.