RでのCycleGanのdiscriminatorについて教えてください。

Question

前提・実現したいこと

以下のサイトから、keras(python)をkeras(R)に書き換えたCycleGanのコードが紹介されているのですが、
discriminatorのコードが書かれておらず、どのように書いたらいいのか教えていただきたいです。

https://github.com/LeeGyeongTak/KerasR/blob/master/CycleGAN.R

発生している問題・エラーメッセージ

DCGANのdiscriminatorの部分を挿入してみましたが、以下のようにエラーが出てしまいました。
Error in dim(valid) <- dim(ak) :
dims [product 20] do not match the length of object [1280]

dimを確認すると、
dim(ak):20 1
dim(valid): 8 8 1 20

であることは分かったのですが、どのようにコードを変更すれば、次元を同じにできるのかわかりません。