code: def create_model(
      feat_size, num_classes, num_layers, model_backend="dgl"
  ):
      model = GNN(
          feat_size,
          64,
          num_classes,
          num_layers,
          model_backend=model_backend,
      )
      model = model.to("cuda")
      model.train()
      return model
  
file_path: cugraph/benchmarks/cugraph-dgl/scale-benchmarks/model.py
imports: import torch
  import torch.nn.functional
  import time
label_desc: Use image node_id 1 for calling a global function with example usage: create_model(feat_size, num_classes, num_layers, model_backend) and returns: model