VAE-GANのサンプルコードはどうやったら動くのでしょう

Question

実現したいこと

Githubの6.0-VAE-GAN-fashion-mnist.ipynbをそのまま実行しようとしても動きません。(tensorflowのバージョン違い？）

Google colabで6.0-VAE-GAN-fashion-mnist.ipynb(https://colab.research.google.com/github/timsainb/tensorflow2-generative-models/blob/master/6.0-VAE-GAN-fashion-mnist.ipynb#scrollTo=HQAi3QxwTScl)を順番に実行してゆくとmodel作成時に
エラーが出ます。

Comments

[jbpb0]

pythonのコードの一番最初の行のすぐ上に
```python
だけの行を追加してください
また、pythonのコードの一番最後の行のすぐ下に
```
だけの行を追加してください

または、
https://teratail.storage.googleapis.com/uploads/contributed_images/56957fe805d9d7befa7dba6a98676d2b.gif
を見て、そのようにしてみてください

現状、コードがとても読み辛いです
質問にコードを載せる際に上記をやってくれたら、他人がコードを読みやすくなり、コードの実行による現象確認もやりやすくなるので、回答されやすくなります

Answer 1

class VAEGAN(tf.keras.Model):
    """a VAEGAN class for tensorflow
    
    Extends:
        tf.keras.Model
    """

    def __init__(self, **kwargs):
        super(VAEGAN, self).__init__()
        self.__dict__.update(kwargs)

        self.enc = tf.keras.Sequential(self.enc)
        self.dec = tf.keras.Sequential(self.dec)
        inputs, disc_l, outputs = self.vae_disc_function()
        self.disc = tf.keras.Model(inputs=[inputs], outputs=[outputs, disc_l])

+       self.D_prop = 1

        self.enc_optimizer = tf.keras.optimizers.Adam(self.lr_base_gen, beta_1=0.5)
        self.dec_optimizer = tf.keras.optimizers.Adam(self.lr_base_gen, beta_1=0.5)
        self.disc_optimizer = tf.keras.optimizers.Adam(self.get_lr_d, beta_1=0.5)

Comments

[hyo-medsoh]

ありがとうございます。　エラーなく学習始めました。

[ps_aux_grep]

ならよかったです．解決済みにしてください．
また，同様のエラーに直面した人のためにも，質問のコメントで指摘されているコードブロックの利用をするように質問を編集してください．