# 需要导入模块: import torch [as 别名]# 或者: from torch importDevice[as 别名]def__init__(self, nn, decoder=kraken.lib.ctc_decoder.greedy_decoder, train: bool = False, device: str ='cpu')->None:""" Constructs a sequence recognizer from a VGSL model and a decoder. Args: nn (kra...
示例1: greedy_decode ▲点赞 6▼ # 需要导入模块: import torch [as 别名]# 或者: from torch importmax[as 别名]defgreedy_decode(self, latent, max_len, start_id):''' latent: (batch_size, max_src_seq, d_model) src_mask: (batch_size, 1, max_src_len) '''batch_size = latent.size...