fix

MaastrichtU-IDS · Jan 26, 2024 · 0445ee5 · 0445ee5
1 parent 7b23440
commit 0445ee5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/train.py b/src/train.py
@@ -391,7 +391,7 @@ def train_gpu(
         y_train, y_test = y[train_index], y[test_index]
 
         # # Send data to GPU for xgboost
-        # send_time = time.time()
+        send_time = time.time()
         # dtrain = xgb.DMatrix(X_train, label=y_train)
         # dtest = xgb.DMatrix(X_test, label=y_test)
         # print(f"Sending data to GPU took {time.time() - send_time}s")