don't plot bias layers

This commit is contained in:
Anton Lydike 2024-11-19 10:10:02 +00:00
parent 94d3a1d484
commit 7861133463

View File

@ -178,6 +178,8 @@ class ExperimentBuilder(nn.Module):
"""
for name, param in named_parameters:
if 'bias' in name:
continue
# Check if the parameter requires gradient and has a gradient
if param.requires_grad and param.grad is not None:
try: