Skip to content

Commit ffb0e90

Browse files
Update simple_cerebros_random_search.py
Try adding a global progress bar.
1 parent 69d9d1d commit ffb0e90

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

cerebros/simplecerebrosrandomsearch/simple_cerebros_random_search.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -518,8 +518,11 @@ def run_moity_permutations(self, spec, subtrial_number, lock):
518518
return 0
519519

520520
def run_random_search(self):
521+
iter_trial = 1
521522
processes = []
522-
for i in np.arange(self.number_of_architecture_moities_to_try):
523+
for i in tqdm(np.arange(self.number_of_architecture_moities_to_try),
524+
desc=f"Moiety {iter_trial}" of {self.number_of_architecture_moities_to_try} running",
525+
colour="#16ceeb"):
523526
self.parse_neural_network_structural_spec_random()
524527
spec = self.get_neural_network_spec()
525528

@@ -537,6 +540,7 @@ def run_random_search(self):
537540
p.start()
538541
for p in processes:
539542
p.join()
543+
iter_trial += 1
540544
# final_oracles = pd.concat(oracles, ignore_index=False)
541545
# if self.direction == "maximize":
542546
# return float(final_oracles[self.metric_to_rank_by].values.max())

0 commit comments

Comments
 (0)