You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by zh...@apache.org on 2020/09/21 19:50:55 UTC
[incubator-mxnet] branch v1.x updated: Tweeking syntax to be closer
to other tests (#19186)
This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch v1.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.x by this push:
new a76cc20 Tweeking syntax to be closer to other tests (#19186)
a76cc20 is described below
commit a76cc20e1244aa61d0d0a5e6b8a0a98e6e9ca03c
Author: Dick Carter <dc...@nvidia.com>
AuthorDate: Mon Sep 21 12:49:59 2020 -0700
Tweeking syntax to be closer to other tests (#19186)
---
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)
})