don't plot bias layers
This commit is contained in:
parent
94d3a1d484
commit
7861133463
@ -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:
|
||||
|
Loading…
Reference in New Issue
Block a user