Expected a default value of type Tensor (inferred) on parameter “alpha“
使用torch.jit.script转的时候，出现错误
Expected a default value of type Tensor (inferred) on parameter “alpha”.Because “alpha” was not annotated with an explicit type it is assumed to be type ‘Tensor’.:
解决
def token_sigmoid_binary_focal_loss(pred_logits, targets, alpha = 0.25, gamma =2.0, text_mask=None, reduction=True):
将函数参数里的类型明确一下：
def token_sigmoid_binary_focal_loss(pred_logits, targets, alpha:float= 0.25, gamma:float =2.0, text_mask=None, reduction:bool=True):本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！