removed some extraneous code related to working on CUDA devices

dhpollack · Jan 6, 2020 · 5db41e8 · 5db41e8
1 parent 464c0a9
commit 5db41e8
Showing 1 changed file with 0 additions and 2 deletions.
diff --git a/src/run_model.cpp b/src/run_model.cpp
@@ -69,8 +69,6 @@ int main(int argc, char *argv[]) {
     cout << "Batch Size: " << mb.data.size(0) << ", " << mb.data.size(1)
          << endl;
     labels_vec.push_back(mb.target);
-    mb.data.to(device);
-    mb.target.to(device);
     auto token_ids = torch::select(mb.data, 1, 0);
     auto attention_masks = torch::select(mb.data, 1, 1);
     auto token_type_ids = torch::select(mb.data, 1, 2);