Bernstein_Poster_2024/basis_nnmf_groups5/append_parameter.py
David Rotermund a540a3f271 Initial
2024-10-21 16:43:42 +02:00

8 lines
193 B
Python

import torch
def append_parameter(
module: torch.nn.Module, parameter_list: list[torch.nn.parameter.Parameter]
):
for netp in module.parameters():
parameter_list.append(netp)