code: def create_model(flags):
      model = Net(use_lstm=flags.use_lstm)
      model.to(device=flags.device)
      return model
  
file_path: moolib/examples/atari/models.py
imports: import torch
  from torch import nn
  import torch.nn.functional
  from moolib.examples.common import nest
label_desc: Use image node_id 1 for calling a global function with example usage: create_model(flags) and returns: model