Update README.md

README.md

@@ -19,7 +19,7 @@ output = model.generate(**tokenizer(text, return_tensors="pt", add_special_token
 tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spaces=True)
 # Output: ' Das ist meine Katze.'
 ```
-(remember the `trust_remote_code=True` because of custom modeling fiel)
+(remember the `trust_remote_code=True` because of custom modeling file)
 ## Training:
 | Parameter            | Value                                                                                           |
 |----------------------|-------------------------------------------------------------------------------------------------|
@@ -31,4 +31,8 @@ tokenizer.decode(output[0], skip_special_tokens=True, clean_up_tokenization_spac
 | Effective Batch Size | 128 (16 * 8)                                                                                    |
 | Training Script      | [train.py](https://github.com/ubaada/scratch-transformer/blob/main/train.py)             |
 | Optimiser            | Adam (learning rate = 0.0001)                                                                   |
+| Loss Type            | Cross Entropy |
+| Final Test Loss      | 1.9 |
+| GPU.                 | RTX 4070 (12GB) |
+