During the test, it said that the shape of the model did not match.

[gaoda1024]

--- load model ---
Traceback (most recent call last):
File "test.py", line 61, in
main()
File "test.py", line 29, in main
model.resume(opts.resume, train=False)
File "/home/gaoda/AutoColor/HistAuGAN/histaugan/model.py", line 444, in resume
self.enc_a.load_state_dict(checkpoint['enc_a'])
File "/home/gaoda/anaconda3/envs/HAG/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1052, in load_state_dict
self.class.name, "\n\t".join(error_msgs)))
RuntimeError: Error(s) in loading state_dict for MD_E_attr_concat:
size mismatch for conv.1.weight: copying a param with shape torch.Size([64, 12, 4, 4]) from checkpoint, the shape in current model is torch.Size([64, 6, 4, 4]).