ynakashima
commited on
Commit
•
b997ba9
1
Parent(s):
850e314
Update README.md
Browse files
README.md
CHANGED
@@ -106,5 +106,5 @@ tokenizer.decode(outputs[0][input_ids.shape[-1]:])
|
|
106 |
|
107 |
### Training Infrastructure
|
108 |
|
109 |
-
- **Hardware**:
|
110 |
- **Software**: We use code based on [neuronx-nemo-megatron](https://github.com/aws-neuron/neuronx-nemo-megatron).
|
|
|
106 |
|
107 |
### Training Infrastructure
|
108 |
|
109 |
+
- **Hardware**: The model was trained on single node of an Amazon EC2 trn1.32xlarge instance.
|
110 |
- **Software**: We use code based on [neuronx-nemo-megatron](https://github.com/aws-neuron/neuronx-nemo-megatron).
|