  import time
  
  from json import JSONDecodeError
  
  import keras
  from keras import losses
  from keras import backend as K
  from keras.models import load_model, Model
  from keras.optimizers import Adam
 
</s> remove     def save_models(self):
 </s> add     def save_models(self, snapshot_iteration): </s> remove     def save(self, should_backup=False):
 </s> add     def save(self, should_backup=False, snapshot=False): </s> remove         print("\n")
 </s> add 
from shutil import copyfile, copytree
  import time
  
  from json import JSONDecodeError
from shutil import copyfile, copytree
  
  import keras
  from keras import losses
  from keras import backend as K
  from keras.models import load_model, Model
  from keras.optimizers import Adam
keep keep add keep keep keep keep keep keep
 Add save snapshot backup option for training 
plugins/train/model/_base.py