Skip to content

Commit c659a7b

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Fix inconsistency in phase_i_b_epochs param in optuna.
1 parent 052e521 commit c659a7b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

generative-proof-of-concept-CPU-preprocessing-in-memory.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -119,7 +119,7 @@ def objective(trial: optuna.Trial) -> float:
119119

120120

121121
epochs = trial.suggest_int('epochs', 50, 75)
122-
phase_i_b_epochs = trial.suggest_int('epochs', 50, 150)
122+
phase_i_b_epochs = trial.suggest_int('phase_i_b_epochs', 50, 150)
123123

124124
batch_size = 5 # trial.suggest_int('batch_size', 5, 10)
125125

0 commit comments

Comments
 (0)