arxyzan commited on
Commit
8e24f26
1 Parent(s): 8a4e849

Hezar: Upload training files

Browse files
Files changed (1) hide show
  1. train/train_config.yaml +21 -0
train/train_config.yaml ADDED
@@ -0,0 +1,21 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ config_type: trainer
2
+ device: cuda
3
+ init_weights_from: hezarai/bert-base-fa
4
+ num_dataloader_workers: 0
5
+ seed: 42
6
+ optimizer:
7
+ name: adam
8
+ config_type: OPTIMIZER
9
+ lr: 2.0e-05
10
+ weight_decay: 0.0
11
+ scheduler:
12
+ name: reduce_on_plateau
13
+ config_type: LR_SCHEDULER
14
+ verbose: true
15
+ batch_size: 8
16
+ use_amp: false
17
+ metrics:
18
+ - seqeval
19
+ num_epochs: 5
20
+ save_freq: 1
21
+ checkpoints_dir: checkpoints/