From 78611334637b8c51c82a977953fb34140f95fc34 Mon Sep 17 00:00:00 2001 From: Anton Lydike Date: Tue, 19 Nov 2024 10:10:02 +0000 Subject: [PATCH] don't plot bias layers --- pytorch_mlp_framework/experiment_builder.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/pytorch_mlp_framework/experiment_builder.py b/pytorch_mlp_framework/experiment_builder.py index 447f62a..e3d0146 100644 --- a/pytorch_mlp_framework/experiment_builder.py +++ b/pytorch_mlp_framework/experiment_builder.py @@ -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: