lukasruff/Deep-SVDD-PyTorch

训练自己数据集时，训练自编码器时的loss特别大，但是在下降，请问可能是什么原因

yokings opened this issue · 5 comments

yokings commented
训练自己数据集时，训练自编码器时的loss特别大，但是在下降，请问可能是什么原因
dbbbbm commented

我一开始也遇到类似问题。第一个原因可能是作者的自编码器输出有sigmoid函数，所以你的输入图像像素值必须在[0,1]之间；第二可能是作者计算loss是求和不是平均，所以看起来大也正常。

  • 👍1
qqqhhh-any commented

作者自定义的loss不是逐像素的均方误差，是求和的

xinjiang0526 commented

所以这loss高的话，需要管他吗，兄弟们

bethansy commented

所以这loss高的话，需要管他吗，兄弟们

请问怎么训练自己的数据哇，我看代码里就限定了'mnist' 和 'cifar10'。我也想换成自己的数据训练后用来预测

bethansy commented

我一开始也遇到类似问题。第一个原因可能是作者的自编码器输出有sigmoid函数，所以你的输入图像像素值必须在[0,1]之间；第二可能是作者计算loss是求和不是平均，所以看起来大也正常。

怎么训练自己的数据呢