config.py配置如何生效
收藏
浏览源码,发现在PaddleSeg/pdseg/utils/config.py /中有许多关于模型的设置,这些设置是否生效?更改这个配置文件有什么作用?
另外,想使用LOVASZ_HINGE_LOSS 测试下效果,在训练时可以用model = pdx.seg.HRNet(num_classes=num_classes, use_dice_loss=True)来使用dice loss,那么LOVASZ loss怎么调用?
0
收藏
请登录后评论
官方说明https://gitee.com/paddlepaddle/PaddleSeg/blob/develop/docs/lovasz_loss.md
在paddleSeg/config中找到你要用的yaml,在SOLVER下面写上LOSS: ['lovasz_hinge_loss','bce_loss']
这样:
SOLVER:
LOSS: ['lovasz_hinge_loss','bce_loss']