Text Generation
Transformers
PyTorch
bloom
text-generation-inference
manojpreveen commited on
Commit
cf5b212
·
1 Parent(s): c8c8d62

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -8,8 +8,9 @@ First Version of Instruction Tuned Bloomz-7B1 model on ChatGPT dataset (85k data
8
  **Base Model:** bigscience/bloomz-7b1
9
 
10
  **Training Details :**
11
-
12
- Epochs: 5
 
13
 
14
  **Dataset Details :**
15
 
 
8
  **Base Model:** bigscience/bloomz-7b1
9
 
10
  **Training Details :**
11
+ * Epochs: 5
12
+ * Batch Size : 5 instantaneous per device x 2 gradient accumulation steps x 8 gpus = 80
13
+ * Machine : 8xA100 80GB
14
 
15
  **Dataset Details :**
16