home
10import copy 11 12from torch import nn 13 14from labml_helpers.module import Module
nn.ModuleList
17def clone_module_list(module: Module, n: int):
21 return nn.ModuleList([copy.deepcopy(module) for _ in range(n)])