現在WGANを一から作っていて,躓いたところがあります.
Real_DataをDiscriminatorに入力したときの出力がD_realとし,
Fake_DataをDiscriminatorに入力したときの出力がD_fakeとしたときのLoss(L_dis)の計算を
L_dis = F.sum(D_real - D_fake)/batch_size
で行いました.
そしてL_dis.backward()を行うとエラーが起きてしまいます.
エラー内容
ValueError: On entry to SGEMV parameter number 8 had an illegal value
調べてもよく分かりませんでした.
Lossの計算が間違っているのでしょうか.