code: def set_layer_dtype(self, _dtype):
      for layer in self.layers:
          for param in layer.parameters():
              param.data = param.data.to(_dtype)
  
imports: import argparse
  import time
  import dgl.graphbolt
  import dgl.nn
  import torch
  import torch.nn
  import torch.nn.functional
  import torchmetrics.functional
  from tqdm import tqdm
label_desc: Use image node_id 2 for calling the SAGE obj's underlying member method code with example usage: obj.set_layer_dtype(_dtype) without return types