enable non-verbose mode in trainer and tester (dynamicslab#124)

Co-authored-by: Hamshaw, Scott Douglas <shamshaw@usgs.gov>

neuralhydrology/evaluation/tester.py

@@ -64,6 +64,8 @@ def __init__(self, cfg: Config, run_dir: Path, period: str = "test", init_model:
         if self.init_model:
             self.model = get_model(cfg).to(self.device)
 
+        self._disable_pbar = cfg.verbose == 0
+
         # pre-initialize variables, defined in class methods
         self.basins = None
         self.scaler = None
@@ -190,7 +192,7 @@ def evaluate(self,
 
         results = defaultdict(dict)
 
-        pbar = tqdm(basins, file=sys.stdout)
+        pbar = tqdm(basins, file=sys.stdout, disable=self._disable_pbar)
         pbar.set_description('# Validation' if self.period == "validation" else "# Evaluation")
 
         for basin in pbar:

neuralhydrology/training/basetrainer.py

@@ -50,6 +50,7 @@ def __init__(self, cfg: Config):
         self._target_std = None
         self._scaler = {}
         self._allow_subsequent_nan_losses = cfg.allow_subsequent_nan_losses
+        self._disable_pbar = cfg.verbose == 0
 
         # load train basin list and add number of basins to the config
         self.basins = load_basin_file(cfg.train_basin_file)
@@ -273,7 +274,7 @@ def _train_epoch(self, epoch: int):
         self.experiment_logger.train()
 
         # process bar handle
-        pbar = tqdm(self.loader, file=sys.stdout)
+        pbar = tqdm(self.loader, file=sys.stdout, disable=self._disable_pbar)
         pbar.set_description(f'# Epoch {epoch}')
 
         # Iterate in batches over training set