class_docstr: Accelerator is an abstraction for the hardware device that is used to run the model.
  
  Args:
      device (str): The device to be used. Currently only support 'cpu' and 'gpu'.
class_docstr_tok: ["Accelerator","is","an","abstraction","for","the","hardware","device","that","is","used","to","run","the","model",".","Args",":","device","(","str",")",":","The","device","to","be","used",".","Currently","only","support","'cpu","'","and","'gpu","'","."]
docstr: Move the model to the device.
  
  Args:
      model (nn.Module): The model to be moved.
docstr_tok: ["Move","the","model","to","the","device",".","Args",":","model","(","nn.Module",")",":","The","model","to","be","moved","."]
code: def configure_model(self, model: nn.Module) -> nn.Module:
      """
      Move the model to the device.
  
      Args:
          model (nn.Module): The model to be moved.
      """
      model = model.to(torch.device(self.device))
      return model
  
imports: import torch
  import torch.nn
label_desc: Use image node_id 3 for calling the Accelerator obj's underlying member method code with example usage: obj.configure_model(model) and returns: model