INVALID_ARGUMENT: required broadcastable shapes while using `tf.distributed.MirroredStrategy()`
I am implementing the VAE model for molecular smiles. (GitHub: https://github.com/dhruv2103/chemvae/tree/tensorflow2). I have made changes to the original git (GitHub: https://github.com/aspuru-guzik-group/chemical_vae) and converted it to run on TensorFlow Version 2.