Skip to content

Commit 94903fa

Browse files
Add experiment config
1 parent 35acf03 commit 94903fa

File tree

1 file changed

+31
-0
lines changed

1 file changed

+31
-0
lines changed
Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,31 @@
1+
# @package _global_
2+
3+
# to execute this experiment run:
4+
# python glm_experiments/train.py experiment=transformer_base
5+
6+
defaults:
7+
- override /data: gpn_animal_promoter
8+
- override /model: dlm_transformer_base
9+
- override /trainer: gpn_animal_promoter
10+
11+
logger:
12+
wandb:
13+
name: experiment-dlm-transformer-base
14+
tags: ["experiment", "dlm", "transformer", "base"]
15+
16+
data:
17+
_target_: glm_experiments.data.lm_datamodule.DLMDataModule
18+
per_device_batch_size: 256
19+
20+
model:
21+
scheduler:
22+
_target_: transformers.get_cosine_with_min_lr_schedule_with_warmup
23+
_partial_: true
24+
num_warmup_steps: 2000
25+
num_training_steps: ${trainer.max_steps}
26+
min_lr_rate: 0.1 # Decay to 10% of max lr
27+
28+
trainer:
29+
max_steps: 20000
30+
log_every_n_steps: 1000
31+
val_check_interval: 1000

0 commit comments

Comments
 (0)