code: def save_embedding(model):
      torch.save(model.embedding.weight.data.cpu(), "embedding.pt")
  
imports: import argparse
  import torch
  from torch_geometric.nn import Node2Vec
  from torch_geometric.utils import to_undirected
  from ogb.nodeproppred import PygNodePropPredDataset
label_desc: Use image node_id 1 for calling a global function with example usage: save_embedding(model) without return types