Update README.md

README.md

@@ -31,7 +31,7 @@ The following hyperparameters were used during training:
 - NUM_EPOCHS = 5
 - MIN_SAVE_EPOCH = 2
 
-The dataset was split 80-10-10 across the train (N=2180), validation (N=272), and test set (N=273). The model was fine-tuned on a single NVIDIA L40 GPU.
+The dataset was split 80-10-10 across the train (N=2180), validation (N=272), and test set (N=273). The model was fine-tuned on a single NVIDIA A40 GPU.
 
 ## How to Get Started with the Model