深度学习 BERT 模型系统班
# 如下为根据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)