  from lib.model.losses import DSSIMObjective, PenalizedLoss
  from lib.model.nn_blocks import NNBlocks
  from lib.multithreading import MultiThread
  from plugins.train._config import Config
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  _CONFIG = None
  
  
 
</s> remove     def save_models(self):
 </s> add     def save_models(self, snapshot_iteration): </s> remove             save_threads.append(MultiThread(network.save, name=name, should_backup=should_backup))
 </s> add             save_threads.append(MultiThread(network.save,
                                            name=name,
                                            should_backup=should_backup))
from lib.utils import get_folder
  from lib.model.losses import DSSIMObjective, PenalizedLoss
  from lib.model.nn_blocks import NNBlocks
  from lib.multithreading import MultiThread
from lib.utils import get_folder
  from plugins.train._config import Config
  
  logger = logging.getLogger(__name__)  # pylint: disable=invalid-name
  _CONFIG = None
  
  
keep keep add keep keep keep keep keep keep
 Add save snapshot backup option for training 
plugins/train/model/_base.py