Cannot find module elastic_weight_consolidation_optimizer on train #288
Description
HyperGAN keeps throwing this error (on colab) when I train hypergan train data/ -c mymodel -s 64x64x3 -b 8 --sample_every 10, or with similar flags.
ModuleNotFoundError: No module named 'hypergan.optimizers.elastic_weight_consolidation_optimizer'
I believe I've installed everything correctly, !hypergan new . -l returns a list of possible configurations.
Steps to reproduce the behaviour
!pip3 install hypergan
I've had to install the following manually
!pip3 install hyperchamber --upgrade --force-reinstall
!pip3 install semantic_version --upgrade --force-reinstall
!hypergan new mymodel -c dcgan
!hypergan train data/ --sampler static_batch -b 8 --config mymodel
Error message
Loading configuration /content/drive/My Drive/Colab Notebooks/stylegan2-ada/unused-data/mymodel.json Creating component: latent Creating component: generator Creating component: discriminator Creating component: loss Creating component: trainer Traceback (most recent call last): File "/usr/local/bin/hypergan", line 141, in gancli.run() File "/usr/local/lib/python3.6/dist-packages/hypergan/cli.py", line 190, in run self.train() File "/usr/local/lib/python3.6/dist-packages/hypergan/cli.py", line 126, in train self.gan = hg.GAN(config=self.gan_config, inputs=self.create_input()) File "/usr/local/lib/python3.6/dist-packages/hypergan/gan.py", line 12, in gan_factory return GANComponent.lookup_function(None, config['class'])(*args, **kw_args) File "/usr/local/lib/python3.6/dist-packages/hypergan/gans/standard_gan.py", line 45, in __init__ BaseGAN.__init__(self, *args, **kwargs) File "/usr/local/lib/python3.6/dist-packages/hypergan/gans/base_gan.py", line 36, in __init__ self.create() File "/usr/local/lib/python3.6/dist-packages/hypergan/gans/standard_gan.py", line 61, in create self.trainer = self.create_component("trainer") File "/usr/local/lib/python3.6/dist-packages/hypergan/gans/base_gan.py", line 99, in create_component gan_component = klass(self, defn, *args, **kw_args) File "/usr/local/lib/python3.6/dist-packages/hypergan/trainers/base_trainer.py", line 10, in __init__ GANComponent.__init__(self, gan, config) File "/usr/local/lib/python3.6/dist-packages/hypergan/gan_component.py", line 36, in __init__ self.create() File "/usr/local/lib/python3.6/dist-packages/hypergan/trainers/base_trainer.py", line 38, in create result = self._create() File "/usr/local/lib/python3.6/dist-packages/hypergan/trainers/alternating_trainer.py", line 13, in _create self.d_optimizer = self.create_optimizer("d_optimizer") File "/usr/local/lib/python3.6/dist-packages/hypergan/trainers/base_trainer.py", line 47, in create_optimizer klass = GANComponent.lookup_function(None, defn['class']) File "/usr/local/lib/python3.6/dist-packages/hypergan/gan_component.py", line 92, in lookup_function return getattr(importlib.import_module(namespace),method) File "/usr/lib/python3.6/importlib/__init__.py", line 126, in import_module return _bootstrap._gcd_import(name[level:], package, level) File "", line 994, in _gcd_import File "", line 971, in _find_and_load File "", line 953, in _find_and_load_unlocked ModuleNotFoundError: No module named 'hypergan.optimizers.elastic_weight_consolidation_optimizer' Error detected, HyperGAN exitingSystem
- Colab, GPU
- Python 3.6.9
- Pip 19.3.1
Configuration File
{ "class": "function:hypergan.losses.standard_loss.StandardLoss", "reduce": "reduce_mean" }, "trainer": { "class": "function:hypergan.trainers.alternating_trainer.AlternatingTrainer", "g_optimizer": { "class": "function:hypergan.optimizers.elastic_weight_consolidation_optimizer.ElasticWeightConsolidationOptimizer",
"f_decay": 0.8,
"optimizer": {
"class": "function:hypergan.optimizers.giga_wolf_optimizer.GigaWolfOptimizer",
"optimizer": {
"class": "function:tensorflow.python.training.adam.AdamOptimizer",
"learn_rate": 1e-4,
"beta1":0.0,
"beta2":0.999
},
"optimizer2": {
"class": "function:tensorflow.python.training.adam.AdamOptimizer",
"learn_rate": 3e-5,
"beta1":0.0,
"beta2":0.999
}
}
},
"d_optimizer": {
"class": "function:hypergan.optimizers.elastic_weight_consolidation_optimizer.ElasticWeightConsolidationOptimizer",
"optimizer": {
"class": "function:hypergan.optimizers.giga_wolf_optimizer.GigaWolfOptimizer",
"optimizer": {
"class": "function:tensorflow.python.training.adam.AdamOptimizer",
"learn_rate": 1e-4,
"beta1":0.0,
"beta2":0.999
},
"optimizer2": {
"class": "function:tensorflow.python.training.adam.AdamOptimizer",
"learn_rate": 3e-5,
"beta1":0.0,
"beta2":0.999
}
}
}
},
"runtime": {
"channels": 3,
"width": 64,
"height": 64,
"train": "hypergan train [dataset] --sampler static_batch -b 8"
},
"hypergan_version": "~1",
"class": "class:hypergan.gans.standard_gan.StandardGAN"
}