Update README.md

federated_learning/README.md

@@ -14,7 +14,7 @@ Below you'll find a step-by-step guide on how to setup and run this project.
 
 In order to run this example as is, you'll need two GPUs on your local machine since each client trains a model independently.
 
-However, if you don't have two GPUs available and would like to train on a single GPU, you can modify the code as follows: in `train-tabformer/custom/tabformer-lstm-trainer.py` on line 84, and in `train-tabformer/custom/tabformer-lstm-validator.py` set `gpu_id = 0`.
+However, if you don't have two GPUs available and would like to train on a single GPU, you can modify the code as follows: in `train-tabformer/custom/tabformer-lstm-trainer.py` on line 84, and in `train-tabformer/custom/tabformer-lstm-validator.py` on line 56 set `gpu_id = 0`.
 
 ## Instructions