欢迎光临散文网 会员登陆 & 注册

深度学习 BERT 模型系统班

2023-01-15 14:16 作者:血霁玫瑰与樱花  | 我要投稿

# 如下为根据nn.Sequential模块中的方法提取而来的,主要帮助我们理解这个模块是怎样执行的# 具体就是将传入的Module一个个按顺序添加起来class MySequential(nn.Module):    from collections import OrderedDict    def __init__(self, *args):        super(MySequential, self).__init__()        if len(args) == 1 and isinstance(args[0], OrderedDict): # 如果传入的是一个OrderedDict            for key, module in args[0].items():                self.add_module(key, module)  

深度学习 BERT 模型系统班的评论 (共 条)

分享到微博请遵守国家法律