refractor

This commit is contained in:
Varuna Jayasiri
2020-09-01 08:12:22 +05:30
parent df22468706
commit 048c5bbd10
3 changed files with 5 additions and 6 deletions

View File

@ -1,9 +0,0 @@
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)])