#复现unet
使用https://github.com/milesial/Pytorch-UNet跑自己的数据集的时候,我的是二分类,我将classes改成2即01分类,照理说没有问题,不存在类别不匹配的情况,但还是出现了标题的错误,于是我想可能是0-255,他并没有归一化成01,所以应该是256.解决
【无标题】unet复现过程错误记载RuntimeError: CUDA error: device-side assert triggered
最新推荐文章于 2023-07-08 21:40:32 发布