Skip to content

Commit 3466b44

Browse files
Add experiment config
1 parent b9270fc commit 3466b44

File tree

1 file changed

+29
-0
lines changed

1 file changed

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

0 commit comments

Comments
 (0)