Skip to content
This repository has been archived by the owner on Feb 9, 2021. It is now read-only.

Commit

Permalink
follow up on fix nightly test (apache#14134)
Browse files Browse the repository at this point in the history
* fix nightly

* trigger ci
  • Loading branch information
roywei authored and Gordon Reid committed Feb 19, 2019
1 parent 1fb87c1 commit be60ec0
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions docs/tutorials/gluon/gluon_from_experiment_to_deployment.md
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,6 @@ Now let's first import necessary packages:
import math
import os
import time
from multiprocessing import cpu_count

from mxnet import autograd
from mxnet import gluon, init
Expand All @@ -77,7 +76,8 @@ lr_factor = 0.75
lr_epochs = [10, 20, 30]

num_gpus = mx.context.num_gpus()
num_workers = cpu_count()
# you can replace num_workers with the number of cores on you device
num_workers = 8
ctx = [mx.gpu(i) for i in range(num_gpus)] if num_gpus > 0 else [mx.cpu()]
batch_size = per_device_batch_size * max(num_gpus, 1)
```
Expand Down

0 comments on commit be60ec0

Please sign in to comment.