search

ganmodokix / gwae

PyTorch Implementation for Gromov-Wasserstein Autoencoders (GWAE)
MIT License
42 stars 4 forks source link

ValueError: Model 'gwae has not been registered' #1

Open pengyanhui opened 4 months ago

pengyanhui commented 4 months ago

When I input the command: python train.py setting/gwae.yaml I got an error: ValueError: Model 'gwae has not been registered' Could you tell me how to register the model 'gwae'?

ganmodokix commented 4 months ago

I tried the command, but it did not reproduce in a new venv environment with pip install -r requirements.txt. Is it sure that import models imports models/__init__.py? If there is a name conflict with it, the gwae model will not be registered.

The model 'gwae' is registered here: https://github.com/ganmodokix/gwae/blob/ee1abfaaa3dadb7fd8a2cf6ccdcec598aa75809e/models/__init__.py#L4

train.py imports the module in question models/__init__.py here: https://github.com/ganmodokix/gwae/blob/ee1abfaaa3dadb7fd8a2cf6ccdcec598aa75809e/train.py#L6