出现报错RuntimeError: mat1 and mat2 shapes cannot be multiplied (57x215296 and 256x100)，不知道是什么问题，向您请教。

[langer-zheng]

您好，感谢您分享的代码！
我在使用自己的训练集运行的时候，出现一下问题，请教您如何解决，谢谢。
100%|██████████| 57/57 [00:00<00:00, 491.31it/s]
0%| | 0/100 [00:00<?, ?it/s]
Traceback (most recent call last):

File "C:\code\WGAN_GP\train_for_vae.py", line 367, in
trainer.train(num_epochs=100,

File "C:\code\WGAN_GP\train_for_vae.py", line 151, in train
recon_loss, kl_diverge = self.compute_batch(batch)

File "C:\code\WGAN_GP\train_for_vae.py", line 196, in compute_batch
outputs, mu, log_var = self.model(images)

File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)

File "C:\code\WGAN_GP\train_for_vae.py", line 88, in forward
mu, log_var = self.encoder(x)

File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)

File "C:\code\WGAN_GP\model\vaemodel.py", line 99, in forward
mu, log_var = self.mu(outputs), self.log_var(outputs)

File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 1110, in _call_impl
return forward_call(*input, **kwargs)

File "C:\ProgramData\Anaconda3\lib\site-packages\torch\nn\modules\linear.py", line 103, in forward
return F.linear(input, self.weight, self.bias)

RuntimeError: mat1 and mat2 shapes cannot be multiplied (57x215296 and 256x100)

[Eric-is-good]

可能的问题
1.如果你直接下载本代码，那么默认vae是可以用的，图片大小必须为 64 * 64
2.你的报错显示 57x215296 ，215296 = 464 * 464，所以我猜测你使用的不是 64 * 64 的图片，你可能用的 464 * 464 的图片
3.请使用 utils.py 的 read_path 来转换图片大小