milesial/Pytorch-UNet

为什么预测出来的图片全黑，能不能回复一下

FRC1221 opened this issue · 5 comments

FRC1221 commented
为什么预测出来的图片全黑，能不能回复一下
huyingjiao commented

你好，我遇到了相同的问题，你的问题解决了吗

FRC1221 commented

你好，我遇到了相同的问题，你的问题解决了吗

没有，换其他的

HunterTom94 commented

你是不是 跑train的时候用了 --amp？可以试试不要用amp再train一遍

deeperlearner commented

I have the same problem. I did not use amp. Did you solve it?

deeperlearner commented

In my case, maybe smaller batch-size can solve it.