wyzhang-ustb / Predictive-and-Adaptive-Deep-Coding-for-Wireless-Image-Transmission-in-Semantic-Communication

Pytorch code for IEEE TWC paper "Predictive and Adaptive Deep Coding for Wireless Image Transmission in Semantic Communication"
97 stars 12 forks source link

train的问题 #7

Open Luckwjf opened 6 months ago

Luckwjf commented 6 months ago

老师您好,我在训练的时候感觉有点不对 lr:1.0000e-04 Epoch: [0][0/391] Loss 0.2502 Epoch: [0][150/391] Loss 0.0000 这样是正常的

zhushumin commented 6 months ago

老师您好,我在训练的时候感觉有点不对 lr:1.0000e-04 Epoch: [0][0/391] Loss 0.2502 Epoch: [0][150/391] Loss 0.0000 这样是正常的 你好,我也遇到了同样的问题,然后这样的模型测试结果都是INF,想要问你解决了吗?

zhushumin commented 6 months ago

老师您好,我在训练的时候感觉有点不对 lr:1.0000e-04 Epoch: [0][0/391] Loss 0.2502 Epoch: [0][150/391] Loss 0.0000 这样是正常的

你好,我想你遇到了同我一样的问题,目前我已经解决,您可以检查util.py文件你16行你所读取的数据是否已经做过归一化,若有,则后续处理不需要再除以 /255, 详情如截图所示 solution

Luckwjf commented 6 months ago

老师您好,我在训练的时候感觉有点不对 lr:1.0000e-04 Epoch: [0][0/391] Loss 0.2502 Epoch: [0][150/391] Loss 0.0000 这样是正常的

你好,我想你遇到了同我一样的问题,目前我已经解决,您可以检查util.py文件你16行你所读取的数据是否已经做过归一化,若有,则后续处理不需要再除以 /255, 详情如截图所示 solution

多谢你的解答,然后我之前在训练的时候也遇到了通道数不匹配问题然后我做了如下修改,不知道这样是否正确。如下图 RuntimeError: Given groups=1, weight of size [256, 3, 5, 5], expected input[256, 32, 3, 32] to have 3 channels, but got 32 channels instead image 希望可以和你多交流交流

zhushumin commented 6 months ago

老师您好,我在训练的时候感觉有点不对 lr:1.0000e-04 Epoch: [0][0/391] Loss 0.2502 Epoch: [0][150/391] Loss 0.0000 这样是正常的

你好,我想你遇到了同我一样的问题,目前我已经解决,您可以检查util.py文件你16行你所读取的数据是否已经做过归一化,若有,则后续处理不需要再除以 /255, 详情如截图所示 solution

多谢你的解答,然后我之前在训练的时候也遇到了通道数不匹配问题然后我做了如下修改,不知道这样是否正确。如下图 RuntimeError: Given groups=1, weight of size [256, 3, 5, 5], expected input[256, 32, 3, 32] to have 3 channels, but got 32 channels instead image 希望可以和你多交流交流

目前来看你这么做也没什么问题,当然可以交流,方便的话微信:13585170511