learning/annotated_deep_learning_paper_implementations
1
0
Fork 0
mirror of https://github.com/labmlai/annotated_deep_learning_paper_implementations.git synced 2025-08-18 11:52:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

refractor

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

9
labml_nn/utils.py Normal file
View File

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