code: def test():
      with monit.section("Create model"):
          model = Model()
      with monit.section("Create probe"):
          probe = ModelProbe(model)
  
      with monit.section("Forward"):
          res = model(torch.tensor([0.1]))
  
      inspect(res)
      inspect(probe.forward_input)
      inspect(probe.forward_output.deep()["l*"].deep().get_dict())
      inspect(probe.parameters["*.bias"].get_dict())
  
      inspect(probe.backward_output)
      inspect(probe.backward_input)
  
      res.sum().backward()
      inspect(probe.backward_output.get_dict())
      inspect(probe.backward_input.get_dict())
  
file_path: labml/client/test/test_model_probe.py
imports: import torch
  from labml import monit
  from labml.internal.analytics.models import ModelProbe
  from labml.logger import inspect
  from torch import nn
label_desc: Use image node_id 1 for calling a global function with example usage: test() without return types