Bernstein_Poster_2024/avg_pooling_mlp_fmask/append_parameter.py
David Rotermund 12eb04e446 New Sims
2024-11-05 18:20:02 +01: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)