gchhablani commited on
Commit
0650f6b
1 Parent(s): 3303276

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -0
README.md CHANGED
@@ -48,6 +48,29 @@ More information needed
48
 
49
  ## Training procedure
50
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
51
  ### Training hyperparameters
52
 
53
  The following hyperparameters were used during training:
 
48
 
49
  ## Training procedure
50
 
51
+ This model is trained using the [run_glue](https://github.com/huggingface/transformers/blob/master/examples/pytorch/text-classification/run_glue.py) script. The following command was used:
52
+
53
+ ```bash
54
+ #!/usr/bin/bash
55
+
56
+
57
+ python ../run_glue.py \
58
+ --model_name_or_path bert-base-cased \
59
+ --task_name sst2 \
60
+ --do_train \
61
+ --do_eval \
62
+ --max_seq_length 512 \
63
+ --per_device_train_batch_size 16 \
64
+ --learning_rate 2e-5 \
65
+ --num_train_epochs 3 \
66
+ --output_dir bert-base-cased-finetuned-sst2 \
67
+ --push_to_hub \
68
+ --hub_strategy all_checkpoints \
69
+ --logging_strategy epoch \
70
+ --save_strategy epoch \
71
+ --evaluation_strategy epoch \
72
+ ```
73
+
74
  ### Training hyperparameters
75
 
76
  The following hyperparameters were used during training: