回帰用のCNNの学習中に起きるエラーの解決法

小熊塁と申します。

タイトルの通り、回帰用のCNNを学習している時にエラーが発生します。

大方の構成は「回帰用の畳み込みニューラルネットワークの学習」を参考にしています。

発生するエラーは

「エラー: trainNetwork (行 184)　全結合層の入力データが無効です。入力データのチャネル数 (1) は、層に必要な入力サイズ (32) と一致しなければなりません。」

というものです。

全結合層の直前にある dropoutLayer を消去することで、このエラーは発生しなくはなるのですが、そうすると今度はRMSEがX軸にめり込んでRMSEの描画が止まってしまい、最終的な結果が出てこず難儀しています。

可能であれば上記のエラーを解決することで学習を完遂したいのですが、アドバイスや対策などあればご教授ください。

コードは、YTrainとYValidationを打ち込む形になってしまった関係上、余りにも長いので質問に全文は載せずに添付という形で共有します。

よろしくお願いします。