Delete gradient for inference

Author: koba-jon

Object_Detection/YOLOv1/src/demo.cpp

@@ -81,6 +81,7 @@ void demo(po::variables_map &vm, torch::Device &device, YOLOv1 &model, std::vect
     std::cout << std::endl;
 
     // (5) Demo
+    torch::NoGradGuard no_grad;
     model->eval();
     flag = true;
     fps = 0.0;

Object_Detection/YOLOv1/src/detect.cpp

@@ -62,6 +62,7 @@ void detect(po::variables_map &vm, torch::Device &device, YOLOv1 &model, std::ve
     std::vector<std::tuple<unsigned char, unsigned char, unsigned char>> label_palette = detector.get_label_palette();
 
     // (4) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["detect_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/detect.txt", std::ios::out);

Object_Detection/YOLOv1/src/test.cpp

@@ -64,6 +64,7 @@ void test(po::variables_map &vm, torch::Device &device, YOLOv1 &model, std::vect
     ave_time = 0.0;
 
     // (5) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["test_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/loss.txt", std::ios::out);

Object_Detection/YOLOv1/src/valid.cpp

@@ -37,6 +37,7 @@ void valid(po::variables_map &vm, DataLoader::ImageFolderBBWithPaths &valid_data
     std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> losses;
 
     // (1) Tensor Forward per Mini Batch
+    torch::NoGradGuard no_grad;
     model->eval();
     iteration = 0;
     total_loss_coord_xy = 0.0; total_loss_coord_wh = 0.0; total_loss_obj = 0.0; total_loss_noobj = 0.0; total_loss_class = 0.0;

Object_Detection/YOLOv2/src/demo.cpp

@@ -81,6 +81,7 @@ void demo(po::variables_map &vm, torch::Device &device, YOLOv2 &model, std::vect
     std::cout << std::endl;
 
     // (5) Demo
+    torch::NoGradGuard no_grad;
     model->eval();
     flag = true;
     fps = 0.0;

Object_Detection/YOLOv2/src/detect.cpp

@@ -62,6 +62,7 @@ void detect(po::variables_map &vm, torch::Device &device, YOLOv2 &model, std::ve
     std::vector<std::tuple<unsigned char, unsigned char, unsigned char>> label_palette = detector.get_label_palette();
 
     // (4) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["detect_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/detect.txt", std::ios::out);

Object_Detection/YOLOv2/src/test.cpp

@@ -64,6 +64,7 @@ void test(po::variables_map &vm, torch::Device &device, YOLOv2 &model, std::vect
     ave_time = 0.0;
 
     // (5) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["test_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/loss.txt", std::ios::out);

Object_Detection/YOLOv2/src/valid.cpp

@@ -37,6 +37,7 @@ void valid(po::variables_map &vm, DataLoader::ImageFolderBBWithPaths &valid_data
     std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> losses;
 
     // (1) Tensor Forward per Mini Batch
+    torch::NoGradGuard no_grad;
     model->eval();
     iteration = 0;
     total_loss_coord_xy = 0.0; total_loss_coord_wh = 0.0; total_loss_obj = 0.0; total_loss_noobj = 0.0; total_loss_class = 0.0;

Object_Detection/YOLOv3/src/demo.cpp

@@ -82,6 +82,7 @@ void demo(po::variables_map &vm, torch::Device &device, YOLOv3 &model, std::vect
     std::cout << std::endl;
 
     // (5) Demo
+    torch::NoGradGuard no_grad;
     model->eval();
     flag = true;
     fps = 0.0;

Object_Detection/YOLOv3/src/detect.cpp

@@ -63,6 +63,7 @@ void detect(po::variables_map &vm, torch::Device &device, YOLOv3 &model, std::ve
     std::vector<std::tuple<unsigned char, unsigned char, unsigned char>> label_palette = detector.get_label_palette();
 
     // (4) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["detect_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/detect.txt", std::ios::out);

Object_Detection/YOLOv3/src/test.cpp

@@ -65,6 +65,7 @@ void test(po::variables_map &vm, torch::Device &device, YOLOv3 &model, std::vect
     ave_time = 0.0;
 
     // (5) Tensor Forward
+    torch::NoGradGuard no_grad;
     model->eval();
     result_dir = vm["test_result_dir"].as<std::string>();  fs::create_directories(result_dir);
     ofs.open(result_dir + "/loss.txt", std::ios::out);

Object_Detection/YOLOv3/src/valid.cpp

@@ -38,6 +38,7 @@ void valid(po::variables_map &vm, DataLoader::ImageFolderBBWithPaths &valid_data
     std::tuple<torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor, torch::Tensor> losses;
 
     // (1) Tensor Forward per Mini Batch
+    torch::NoGradGuard no_grad;
     model->eval();
     iteration = 0;
     total_loss_coord_xy = 0.0; total_loss_coord_wh = 0.0; total_loss_obj = 0.0; total_loss_noobj = 0.0; total_loss_class = 0.0;