You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by sa...@apache.org on 2020/09/22 16:59:39 UTC
[incubator-mxnet] branch v1.8.x updated: Tweeking syntax to be
closer to other tests (#19186) (#19206)
This is an automated email from the ASF dual-hosted git repository.
samskalicky pushed a commit to branch v1.8.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.8.x by this push:
new c2df97f Tweeking syntax to be closer to other tests (#19186) (#19206)
c2df97f is described below
commit c2df97faacaa4fcee4c4ee3e79fe4e325023e78e
Author: Dick Carter <dc...@nvidia.com>
AuthorDate: Tue Sep 22 09:55:44 2020 -0700
Tweeking syntax to be closer to other tests (#19186) (#19206)
---
R-package/tests/testthat/test_img_seg.R | 3 +--
1 file changed, 1 insertion(+), 2 deletions(-)
diff --git a/R-package/tests/testthat/test_img_seg.R b/R-package/tests/testthat/test_img_seg.R
index 4af7d62..20d09f6 100644
--- a/R-package/tests/testthat/test_img_seg.R
+++ b/R-package/tests/testthat/test_img_seg.R
@@ -154,12 +154,11 @@ test_that("UNET", {
train.y.array <- train.y
dim(train.y.array) <- c(IMG_SIZE, IMG_SIZE, 1, 30)
- devices <- mx.ctx.default()
mx.set.seed(0)
net <- get_unet()
model <- mx.model.FeedForward.create(net, X = train.array, y = train.y.array,
- ctx = devices, num.round = 2, initializer = mx.init.normal(sqrt(2/576)),
+ ctx = mx.ctx.default(), num.round = 2, initializer = mx.init.normal(sqrt(2/576)),
learning.rate = 0.05, momentum = 0.99, array.batch.size = 2)
})