Fix for more arguments
This commit is contained in:
parent
7a4edb1cd8
commit
e89fd33229
@ -557,7 +557,7 @@ class Hyperopt:
|
||||
cpus = cpu_count()
|
||||
logger.info(f"Found {cpus} CPU cores. Let's make them scream!")
|
||||
config_jobs = self.config.get('hyperopt_jobs', -1)
|
||||
if self.total_epochs < cpus:
|
||||
if self.total_epochs < cpus and (config_jobs > self.total_epochs or config_jobs < 0):
|
||||
config_jobs = self.total_epochs
|
||||
logger.info(f'Number of parallel jobs set as: {config_jobs}')
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user