mirror of
https://github.com/labmlai/annotated_deep_learning_paper_implementations.git
synced 2025-10-31 02:39:16 +08:00
20 lines
373 B
Python
20 lines
373 B
Python
"""
|
|
---
|
|
title: Utilities
|
|
summary: A bunch of utility functions and classes
|
|
---
|
|
|
|
# Utilities
|
|
"""
|
|
|
|
import copy
|
|
|
|
from labml_helpers.module import M, TypedModuleList
|
|
|
|
|
|
def clone_module_list(module: M, n: int) -> TypedModuleList[M]:
|
|
"""
|
|
## Make a `nn.ModuleList` with clones of a given layer
|
|
"""
|
|
return TypedModuleList([copy.deepcopy(module) for _ in range(n)])
|