Files
Varuna Jayasiri 6924f4580c lab_helpers
2020-09-01 08:05:08 +05:30

10 lines
192 B
Python

import copy
from torch import nn
from labml_helpers.module import Module
def clone_module_list(module: Module, n: int):
return nn.ModuleList([copy.deepcopy(module) for _ in range(n)])