uer commited on
Commit
5638fac
1 Parent(s): 95d58da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +11 -3
README.md CHANGED
@@ -63,12 +63,20 @@ python3 pretrain.py --dataset_path poem.pt \
63
  --output_model_path models/poem_gpt_base_model.bin \
64
  --config_path models/bert_base_config.json --learning_rate 5e-4 \
65
  --tie_weight --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
66
- --batch_size 64 --report_steps 1000 \
67
- --save_checkpoint_steps 50000 --total_steps 200000 \
68
- --embedding gpt --encoder gpt2 --target lm
 
69
 
70
  ```
71
 
 
 
 
 
 
 
 
72
  ### BibTeX entry and citation info
73
 
74
  ```
 
63
  --output_model_path models/poem_gpt_base_model.bin \
64
  --config_path models/bert_base_config.json --learning_rate 5e-4 \
65
  --tie_weight --world_size 8 --gpu_ranks 0 1 2 3 4 5 6 7 \
66
+ --embedding word_pos --remove_embedding_layernorm \
67
+ --encoder transformer --mask causal --target lm\
68
+ --layernorm_positioning pre --batch_size 64 --report_steps 1000 \
69
+ --save_checkpoint_steps 50000 --total_steps 200000
70
 
71
  ```
72
 
73
+ Finally, we convert the pre-trained model into Huggingface's format:
74
+ ```
75
+ python3 scripts/convert_gpt2_from_uer_to_huggingface.py --input_model_path poem_gpt_base_model.bin\
76
+ --output_model_path pytorch_model.bin\
77
+ --layers 12
78
+ ```
79
+
80
  ### BibTeX entry and citation info
81
 
82
  ```