          self.stop = True
          thr.join() # waits until thread finishes
      
      def processThread(self):
          variant = "Original" # TODO Pass as argument
          
          print('Loading data, this may take a while...')
          model = PluginLoader.get_model(variant)(self.arguments.model_dir)
          model.load(swapped=False)
  
          images_A = get_image_paths(self.arguments.input_A)
 
</s> remove         model = PluginLoader.get_model(variant)(self.arguments.model_dir)
 </s> add         model = PluginLoader.get_model(self.arguments.trainer)(self.arguments.model_dir) </s> remove         trainer = PluginLoader.get_trainer(variant)(model, images_A, images_B)
 </s> add         trainer = PluginLoader.get_trainer(self.arguments.trainer)(model,
                                                                   images_A,
                                                                   images_B,
                                                                   batch_size=self.arguments.batch_size)
          self.stop = True
          thr.join() # waits until thread finishes
      
      def processThread(self):


          print('Loading data, this may take a while...')
          model = PluginLoader.get_model(variant)(self.arguments.model_dir)
          model.load(swapped=False)
  
          images_A = get_image_paths(self.arguments.input_A)
keep keep keep keep replace replace keep keep keep keep keep
 added batch-size cli argumet.
neatened up train.py and added LowMem trainer option.
Added save_now/show key to preview. 
scripts/train.py