Train VaeEncoderDocder issues

[chenhy-97]

We attempt to train a new encoder, but it's shown this errors as follow.

Traceback (most recent call last):
File "tools/train_encoder.py", line 161, in
main()
File "tools/train_encoder.py", line 138, in main
test_cfg=cfg.test_cfg)
File "/root/picasso/haoyuan_chen/chy/mmgeneration/mmgen/models/builder.py", line 32, in build_model
return build(cfg, MODELS, dict(train_cfg=train_cfg, test_cfg=test_cfg))
File "/root/picasso/haoyuan_chen/chy/mmgeneration/mmgen/models/builder.py", line 27, in build
return build_from_cfg(cfg, registry, default_args)
File "/root/picasso/haoyuan_chen/anaconda3/envs/facestylor/lib/python3.7/site-packages/mmcv/utils/registry.py", line 45, in build_from_cfg
f'{obj_type} is not in the {registry.name} registry')
KeyError: 'PSPEncoderDecoder is not in the model registry'

And we refer to the methods on the forum, such as reconfiguring the requirements.txt and using @models register_ module(name='PSPEncoderDecoder', force=True), but they are useless.

We now suspect that it is caused by the mismatch between the version of mmcv and the version of mmgen. Our version of mmcv is 1.4.8 and the version of mmgen is 0.6.0.