edit bit vision

Atashnezhad · Jun 20, 2023 · edcf522 · edcf522
1 parent c04d862
commit edcf522
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/neural_network_model/bit_vision.py b/neural_network_model/bit_vision.py
@@ -257,16 +257,15 @@ def train_model(
  self.model_history = self.model.fit_generator(
  steps_per_epoch=steps_per_epoch,
  generator=self.train_val_gens["train"],
- epochs=epochs or SETTING.MODEL_SETTING.EPOCHS,
- verbose=verbose or SETTING.MODEL_SETTING.FIT_GEN_VERBOSE,
+ epochs=epochs,
+ verbose=verbose,
  validation_data=self.train_val_gens["val"],
- validation_steps=val_steps or SETTING.MODEL_SETTING.VALIDATION_STEPS,
- class_weight=class_weight or SETTING.MODEL_SETTING.CLASS_WEIGHT,
+ validation_steps=val_steps,
+ class_weight=class_weight,
  max_queue_size=SETTING.MODEL_SETTING.MAX_QUEUE_SIZE,
- workers=workers or SETTING.MODEL_SETTING.WORKERS,
- use_multiprocessing=use_multiprocessing
- or SETTING.MODEL_SETTING.USE_MULTIPROCESSING,
- shuffle=shuffle or SETTING.MODEL_SETTING.SHUFFLE,
+ workers=workers,
+ use_multiprocessing=use_multiprocessing,
+ shuffle=shuffle,
  initial_epoch=SETTING.MODEL_SETTING.INITIAL_EPOCH,
  callbacks=[self._check_points(model_save_address, model_name)],
  )