Libtorch ubuntu runtime error

hi all.
I have a question.

same code runs fine on Windows, but a runtime error occurs on Linux.
Can you look at me together?

this is my repo https://github.com/kerry-Cho/transfer-learning-Libtorch

How to start building

1. run bash build.sh 
2. cd build/
3. make 
4. cd bin/
5. run ./transfer-learning

below crash code

torch::Tensor read_data(std::string location) {
	/*
	 Function to return image read at location given as type torch::Tensor
	 Resizes image to (224, 224, 3)
	 Parameters
	 ===========
	 1. location (std::string type) - required to load image from the location
	 Returns
	 ===========
	 torch::Tensor type - image read as tensor
	*/
	std::cout << location << std::endl;
	cv::Mat img = cv::imread(location, 1);
	cv::resize(img, img, cv::Size(224, 224), cv::INTER_CUBIC);
	torch::Tensor img_tensor = torch::from_blob(img.data, { img.rows, img.cols, 3 }, torch::kByte); << here crash point
	img_tensor = img_tensor.permute({ 2, 0, 1 });
	return img_tensor.clone();
}

@yf225 help me :smile: :smile: :smile:

@yf225 sorry.
It was my mistake.
It happened because of a file path error