code: def model_state(self, value):
      self._model_state = value
  
imports: import warnings
  import psutil
  import torch
  import torch.nn
  from tqdm import tqdm
  from tez import enums
  from tez.callbacks import CallbackRunner
  from tez.utils import AverageMeter
label_desc: Use image node_id 3 for calling the Model obj's underlying member method code with example usage: obj.model_state(value) without return types