Skip to content

Commit 8650500

Browse files
Update ames-wo-val-hyperparam-optimization-local.py
File name correction on deletion of artifacts.
1 parent c2c85ac commit 8650500

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

ames-wo-val-hyperparam-optimization-local.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,8 @@ def objective(trial):
7272
.replace(':', '_')\
7373
.replace('-', '_')
7474
PROJECT_NAME = f'{TIME}_cerebros_auto_ml_test'
75+
project_name_full =\
76+
f"{PROJECT_NAME}_meta_{meta_trial_number}"
7577

7678
cerebros = SimpleCerebrosRandomSearch(
7779
unit_type=DenseUnit,
@@ -112,7 +114,7 @@ def objective(trial):
112114
metrics=[tf.keras.metrics.RootMeanSquaredError()],
113115
epochs=epochs,
114116
patience=7,
115-
project_name=f"{PROJECT_NAME}_meta_{meta_trial_number}",
117+
project_name=project_name_full,
116118
model_graphs='model_graphs',
117119
batch_size=batch_size,
118120
meta_trial_number=meta_trial_number)
@@ -121,7 +123,7 @@ def objective(trial):
121123

122124
# Delete artifacts so we don't deplete disk space:
123125

124-
rmtree(project_name)
126+
rmtree(project_name_full)
125127

126128
# Make a copy of the result so it isn't lost in garbage collection.
127129

0 commit comments

Comments
 (0)