code: def run(self):
      self.model = torch.nn.DataParallel(self.model)
      self.model.to(self.device)
      # train
      self.train()
      # validate
      self.validate(is_test=False)
      # test
      self.validate(is_test=True)
  
file_path: nni/examples/nas/legacy/oneshot/proxylessnas/retrain.py
imports: import time
  import math
  from datetime import timedelta
  import torch
  from torch import nn
  from nni.nas.pytorch.utils import AverageMeter
label_desc: Use image node_id 2 for calling the Retrain obj's underlying member method code with example usage: obj.run() without return types