deep-learning-for-image-processing icon indicating copy to clipboard operation
deep-learning-for-image-processing copied to clipboard
verified publisher icon WZMIAOMIAO

unet训练时问题

Open wangwxr opened this issue 2 years ago • 0 comments

您好，我使用unet在DRIVE数据集内容不变，resize为512*512的大小时运行train.py报错，交叉熵函数Target 244 is out of bounds.请问是不是激活函数设置成了与图像大小相关。如果想修改的话，在哪里可以修改

wangwxr avatar Aug 23 '23 12:08 wangwxr