Skip to content

Commit c716551

Browse files
committed
updated readme to use docker-compose-gpu.yml for gpu training
1 parent 34fa832 commit c716551

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

README.md

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -78,6 +78,10 @@ Move to the directory where you cloned *Sherlock* , and run:
7878
```bash
7979
docker-compose up --build
8080
```
81+
Training using BERT runs much faster on GPU with >12GB RAM (Tested with Nvidia K80). To train with GPU run:
82+
```bash
83+
docker-compose -f docker-compose-gpu.yml up --build
84+
```
8185

8286
If everything goes well, you should start seeing the building message of the docker containers:
8387
```

0 commit comments

Comments
 (0)