We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 3311817 commit 8563b14Copy full SHA for 8563b14
tests/python-sycl/test_sycl_training_continuation.py
@@ -9,11 +9,11 @@ class TestSYCLTrainingContinuation:
9
def run_training_continuation(self, use_json):
10
kRows = 64
11
kCols = 32
12
- X = np.random.randn(kRows, kCols)
13
- y = np.random.randn(kRows)
+ X = rng.randn(kRows, kCols)
+ y = rng.randn(kRows)
14
dtrain = xgb.DMatrix(X, y)
15
params = {
16
- "device": "sycl",
+ "device": "sycl:cpu",
17
"max_depth": "2",
18
"gamma": "0.1",
19
"alpha": "0.01",
0 commit comments