处理已保存模型中的自定义层或其他自定义对象
自己在使用magpie的时候定义了一个loss函数,为了解决类别不平衡的问题
from magpie import Magpie
from magpie.nn.models import weighted_loss
magpie = Magpie()
magpie.load_model('./dj.h5', custom_objects={'weighted_loss': weighted_loss})
处理已保存模型中的自定义层或其他自定义对象
自己在使用magpie的时候定义了一个loss函数,为了解决类别不平衡的问题
from magpie import Magpie
from magpie.nn.models import weighted_loss
magpie = Magpie()
magpie.load_model('./dj.h5', custom_objects={'weighted_loss': weighted_loss})
本文标题:如何load自己定制的loss函数模型
本文链接:https://www.haomeiwen.com/subject/snrtmqtx.html
网友评论