Key error when loading pretrained MCC module

[thunguyenphuoc]

I'm trying to run the prestrained module but get the error when I was trying to load the state_dict for the MCC_Module (mcc_module_iter_160000.pth). The error is as follows:

RuntimeError: Error(s) in loading state_dict for MCC_Module:
Unexpected key(s) in state_dict: "MCCN.h.weight", "MCCN.h.bias".

[XYkong-CS]

We have revised the code and you can solve the problem by updating "net.py".