Hannes Kuchelmeister
commited on
Commit
·
993f17b
1
Parent(s):
353bc80
change docker-compose to not use gpu
Browse files- docker-compose.yml +1 -1
docker-compose.yml
CHANGED
|
@@ -9,7 +9,7 @@ services:
|
|
| 9 |
- ./logs/:/usr/src/app/logs:z
|
| 10 |
- ./data/:/usr/src/app/data:z
|
| 11 |
- ./configs:/usr/src/app/configs:z
|
| 12 |
-
command: sh -c "python train.py logger=tensorboard"
|
| 13 |
tensorboard:
|
| 14 |
image: tensorflow/tensorflow
|
| 15 |
volumes:
|
|
|
|
| 9 |
- ./logs/:/usr/src/app/logs:z
|
| 10 |
- ./data/:/usr/src/app/data:z
|
| 11 |
- ./configs:/usr/src/app/configs:z
|
| 12 |
+
command: sh -c "python train.py logger=tensorboard trainer.gpus=0"
|
| 13 |
tensorboard:
|
| 14 |
image: tensorflow/tensorflow
|
| 15 |
volumes:
|