CNNの全結合層の出力サイズについて

Question

前提・実現したいこと

CNNについてのプログラムを書いているのですが、おそらく出力サイズの部分でエラーが発生してしまいます。
画像のサイズは224*224です。

画像サイズを基に適切な出力サイズを決めるには、どのような計算をすればよいのでしょうか。

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

1

RuntimeError: shape '[-1, 400]' is invalid for input of size 2876416

2

ValueError: Expected input batch_size (1) to match target batch_size (64).

該当のソースコード

1

self.fc1 = nn.Linear(16*5*5, 256)

2

self.fc1 = nn.Linear(16*424*424, 256)

試したこと

最初に試した時のソースコードとエラーを1、
直した結果を2として記載しています。

1でのエラーメッセージで2876416という数字が出ているため、16424424ならちょうど2876416になると思って直したのですが別のエラー内容になりました。