深度學(xué)習(xí) BERT 模型系統(tǒng)班
# 如下為根據(jù)nn.Sequential模塊中的方法提取而來的,主要幫助我們理解這個(gè)模塊是怎樣執(zhí)行的# 具體就是將傳入的Module一個(gè)個(gè)按順序添加起來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): # 如果傳入的是一個(gè)OrderedDict
? ? ? ? ? ?for key, module in args[0].items():
? ? ? ? ? ? ? ?self.add_module(key, module) ?
標(biāo)簽: