code: def test_tracking_one_layer_model(self):
      self.num_layer = 1
      self.pytorch_network = nn.Linear(1, 1)
      self.optimizer = torch.optim.SGD(
          self.pytorch_network.parameters(), lr=self.lr
      )
      self.model = Model(
          self.pytorch_network, self.optimizer, self.loss_function
      )
  
      keep_bias = False
      train_gen = some_data_generator(20)
      valid_gen = some_data_generator(20)
      tracker = TensorBoardGradientTracker(
          self.writer, keep_bias=keep_bias
      )
      self.model.fit_generator(
          train_gen,
          valid_gen,
          epochs=self.num_epochs,
          steps_per_epoch=5,
          callbacks=[tracker],
      )
      self._test_tracking(keep_bias)
  
imports: import math
  from tempfile import TemporaryDirectory
  from unittest import TestCase, skipIf
  from unittest.mock import ANY, MagicMock, call
  import torch
  import torch.nn
  from tests.framework.tools import some_data_generator
  from poutyne import GradientTracker, Model, TensorBoardGradientTracker, WeightsGradientsStatsTracker
label_desc: Use image node_id 3 for calling the TensorBoardGradientTrackerTest obj's underlying member method code with example usage: obj.test_tracking_one_layer_model() without return types