code: def reset_balance_profile(balance_dict, num_layers, balance_strategy):
      for key in metrics:
          balance_dict[key] = [None for _ in range(num_layers)]
      if balance_strategy:
          balance_dict[f"{balance_strategy}_loss"] = [
              None for _ in range(num_layers)
          ]
  
imports: import torch
  import torch.nn.functional
label_desc: Use image node_id 1 for calling a global function with example usage: reset_balance_profile(balance_dict, num_layers, balance_strategy) without return types