Style transfer MPS working for some weird reason. Come back to here.

Iainmon · Iainmon · commit 36dcd2c9345b · 2025-05-04T21:52:43.000-07:00
diff --git a/demos/video/style-transfer/style_transfer.cpp b/demos/video/style-transfer/style_transfer.cpp
@@ -89,7 +89,8 @@ torch::indexing::Slice slice() {
 }
 
 torch::Tensor test_channel(torch::Tensor& input) {
-    std::cout << "Input sizes: " << input.sizes() << std::endl;
+    std::cout << "Input device: " << input.device() << std::endl;
+
     int channel_to_disable = 0;
     // auto img = input.select(1, channel_to_disable).zero();  
     auto output = input.clone();

Original file line number	Diff line number	Diff line change
`@@ -89,7 +89,8 @@ torch::indexing::Slice slice() {`
`89`	`89`	`}`
`90`	`90`
`91`	`91`	`torch::Tensor test_channel(torch::Tensor& input) {`
`92`		`- std::cout << "Input sizes: " << input.sizes() << std::endl;`
	`92`	`+ std::cout << "Input device: " << input.device() << std::endl;`
	`93`	`+`
`93`	`94`	`int channel_to_disable = 0;`
`94`	`95`	`// auto img = input.select(1, channel_to_disable).zero();`
`95`	`96`	`auto output = input.clone();`