Upload TRAIN_motionstreamer.sh
Browse files- TRAIN_motionstreamer.sh +1 -1
TRAIN_motionstreamer.sh
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
NUM_GPUS=${1:-1} # default: 1 GPU
|
| 2 |
|
| 3 |
-
BATCH_SIZE=$((
|
| 4 |
|
| 5 |
echo "Using $NUM_GPUS GPUs, each with a batch size of $BATCH_SIZE"
|
| 6 |
|
|
|
|
| 1 |
NUM_GPUS=${1:-1} # default: 1 GPU
|
| 2 |
|
| 3 |
+
BATCH_SIZE=$((256 / NUM_GPUS))
|
| 4 |
|
| 5 |
echo "Using $NUM_GPUS GPUs, each with a batch size of $BATCH_SIZE"
|
| 6 |
|