Skip to content

Commit acbd68d

Browse files
Rename modules
1 parent 33dcee6 commit acbd68d

File tree

8 files changed

+5
-5
lines changed

8 files changed

+5
-5
lines changed

configs/data/default.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
_target_: glm_experiments.data.dna_datamodule.MLMDataModule
1+
_target_: glm_experiments.data.lm_datamodule.MLMDataModule
22

33
# Dataset and tokenizer (override in specific configs)
44
dataset_name: data/gpn-animal-promoter-dataset

configs/experiment/clm_transformer_base.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ logger:
1414
tags: ["experiment", "clm", "transformer", "base"]
1515

1616
data:
17-
_target_: glm_experiments.data.dna_datamodule.CLMDataModule
17+
_target_: glm_experiments.data.lm_datamodule.CLMDataModule
1818
per_device_batch_size: 256
1919

2020
model:

configs/experiment/clm_transformer_small.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ model:
3030
num_training_steps: ${trainer.max_steps}
3131

3232
data:
33-
_target_: glm_experiments.data.dna_datamodule.CLMDataModule
33+
_target_: glm_experiments.data.lm_datamodule.CLMDataModule
3434
batch_size: 8
3535
per_device_batch_size: 8
3636

configs/experiment/mlm_transformer_base.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55

66
defaults:
77
- override /data: gpn_animal_promoter
8-
- override /model: bert_transformer_base
8+
- override /model: mlm_transformer_base
99
- override /trainer: gpn_animal_promoter
1010

1111
logger:

configs/experiment/transformer_small.yaml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
# Short training run with small Transformer encoder for quick testing
44

55
defaults:
6-
- override /model: bert_transformer_small
6+
- override /model: mlm_transformer_small
77

88
logger:
99
wandb:

0 commit comments

Comments
 (0)