code: def init_params():
      # Initialize lazy parameters via forwarding a single batch to the model:
      batch = next(iter(train_loader))
      batch = batch.to(device, "edge_index")
      model(batch.x_dict, batch.edge_index_dict)
  
file_path: pytorch_geometric/examples/hetero/to_hetero_mag.py
imports: import argparse
  import os.path
  import torch
  import torch.nn.functional
  from torch.nn import ReLU
  from tqdm import tqdm
  import torch_geometric.transforms
  from torch_geometric.datasets import OGB_MAG
  from torch_geometric.loader import HGTLoader, NeighborLoader
  from torch_geometric.nn import Linear, SAGEConv, Sequential, to_hetero
label_desc: Use image node_id 1 for calling a global function with example usage: init_params() without return types