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):