Skip to content

Commit 2aba971

Browse files
committed
Ups Set llm experiment back to default, no hebb, 1e-4 lr
1 parent fb5fdeb commit 2aba971

1 file changed

Lines changed: 2 additions & 3 deletions

File tree

examples/advanced/experiment_llm.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@
6262
# OPTIMIZER CONFIG
6363
RESET_OPTIMIZER_ON_LOAD = False
6464
OVERWRITE_LR_OF_CKPT = True
65-
LEARNING_RATE = 1e-5
65+
LEARNING_RATE = 1e-4
6666

6767
# TIE EMBEDDINGS (VRAM Saving & Parameter Sharing)
6868
TIE_EMBEDDINGS = False
@@ -292,8 +292,7 @@ def initialize_system(vocab_size, num_neurons, device, input_count=-1, output_co
292292
vocab_size=vocab_size,
293293
vocab_mode='discrete',
294294
tie_embeddings=TIE_EMBEDDINGS,
295-
debug=debug,
296-
hebb_type='synapse'
295+
debug=debug
297296
)
298297

299298
trainer = OdyssNetTrainer(

0 commit comments

Comments
 (0)