A little bit confuse with KeyError: 'a' about munit HOT 3 OPEN

hi, i got the problem too, when run the test.py file, did u fix it? @JSharp4273

Traceback (most recent call last):
File "test.py", line 55, in
trainer.gen_a.load_state_dict(state_dict['a'])
KeyError: 'a'

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
File "test.py", line 58, in
state_dict = pytorch03_to_pytorch04(torch.load(opts.checkpoint), opts.trainer)
File "/home/xiang/git/MUNIT/utils.py", line 384, in pytorch03_to_pytorch04
state_dict['a'] = __conversion_core(state_dict_base['a'], trainer_name)
KeyError: 'a'

from munit.

if use gen_0****.pt, it's ok, but what's the dis_0****.pt and optimizer.pt,
"gen_a , auto-encoder for domain a" in trainer.py line17 that i could understand, but "dis_a, discriminator for domain a " in trainer.py line19 that i don't think i can understand, and more the model optimizer.pt just use to help training? and what do model dis_0***.pt for ?
just use the model gen_0***.pt when test ?
anyone can help me? thank you very much

from munit.

I meet the same problem too. I use the gen_*.pt when test.

from munit.