NameError: name 'TerminalGRU' is not defined

[ramashka328]

Hi. I am trying to implement your model using the code line as follows:

vae = VAEUtils(directory='C:\Users\Autoencoder\chemical_vae-master\models\zinc_properties')

but however it results in the error:

~\Autoencoder\chemical_vae-master\chemvae\vae_utils.py in init(self, exp_file, encoder_file, decoder_file, directory)
55 # encoder, decoder
56 self.enc = load_encoder(self.params)
---> 57 self.dec = load_decoder(self.params)
58 self.encode, self.decode = self.enc_dec_functions()
59 self.data = None

~\Autoencoder\chemical_vae-master\chemvae\vae_utils.py in load_decoder(params)
21 def load_decoder(params):
22 if params['do_tgru']:
---> 23 return load_model(params['decoder_weights_file'], custom_objects={'TerminalGRU': TerminalGRU})
24 else:
25 return load_model(params['decoder_weights_file'])

NameError: name 'TerminalGRU' is not defined

Could you tell me, what should I do to make it work?

[mkduong-ai]

Did you install it via setup.py?
tgru_k2_gpu.py is the file where the TerminalGRU Layer is defined for the decoder model. Check whether vae_utils.py has from tgru_k2_gpu import TerminalGRU in it.

[ramashka328]

Thank You, it worked when I installed it using setup.py.