Skip to content

Commit 4c4e0e5

Browse files
Update generative-proof-of-concept-CPU-preprocessing-in-memory.py
Lower embedding dim to fit the vanilla scale CICD workflow.
1 parent 428c0ec commit 4c4e0e5

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
@@ -150,7 +150,7 @@ def objective(trial: optuna.Trial) -> float:
150150
# embedding output dim must be an even number
151151
# Maximize EMBEDDING_N based on available RAM and CPU / GPU
152152

153-
EMBEDDING_N = 9 # trial.suggest_int('embedding_n',6, 9) # 12
153+
EMBEDDING_N = 6 # trial.suggest_int('embedding_n',6, 9) # 12
154154
EMBEDDING_DIM = int(EMBEDDING_N * 2)
155155

156156
PROJECTION_N = 1 # Punatuve increase of ram, leaving this as 1 until we are running on HPC

0 commit comments

Comments
 (0)