Skip to content

Commit

Permalink
Merge pull request NVIDIA#157 from annikabrundyn/main
Browse files Browse the repository at this point in the history
Update readme to include system requirements
  • Loading branch information
esnvidia authored Aug 8, 2022
2 parents 740ef8c + 84f249d commit 94ac1d1
Showing 1 changed file with 6 additions and 0 deletions.
6 changes: 6 additions & 0 deletions federated_learning/README.md
Original file line number Diff line number Diff line change
Expand Up @@ -10,6 +10,12 @@ All our FL code can be found in the `train-tabformer` folder. Inside you'll see

Below you'll find a step-by-step guide on how to setup and run this project.

## Requirements

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` on line 56 set `gpu_id = 0`.

## Instructions

First, you'll need to setup the data and virtual environment following the steps in the file [`0_Setup.md`](/0_Setup.md).
Expand Down

0 comments on commit 94ac1d1

Please sign in to comment.