revert

2Delight · May 20, 2023 · 31c0d7b · 31c0d7b
1 parent 6731586
commit 31c0d7b
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/.github/workflows/macos-test.yaml b/.github/workflows/macos-test.yaml
@@ -28,9 +28,6 @@ jobs:
       - name: Setup Environment
         run: ./scripts/setup.sh
 
-      - name: LTPT
-        run: export LIBTORCH_USE_PYTORCH=1
-
       - name: Test
         run: cargo test --verbose --verbose
         working-directory: src-tauri
diff --git a/src-tauri/Cargo.toml b/src-tauri/Cargo.toml
@@ -23,7 +23,7 @@ simple_logger = "4.0"
 
 rand = "0.8"
 
-tch = "0.13"
+tch = "0.10"
 image = "0.24"
 
 cpal = "0.15"

diff --git a/src-tauri/src/mascot.rs b/src-tauri/src/mascot.rs
@@ -22,9 +22,9 @@ pub struct Mascot {
 ///
 /// It takes original 3-channel tensor and makes 1-channel tensor with mean vallues.
 pub fn to_bw(tensor: Tensor) -> Tensor {
-    (tensor.index(&[Some(Tensor::from_slice(&[0i64])), None, None])
-        + tensor.index(&[Some(Tensor::from_slice(&[1i64])), None, None])
-        + tensor.index(&[Some(Tensor::from_slice(&[2i64])), None, None]))
+    (tensor.index(&[Some(Tensor::of_slice(&[0i64])), None, None])
+        + tensor.index(&[Some(Tensor::of_slice(&[1i64])), None, None])
+        + tensor.index(&[Some(Tensor::of_slice(&[2i64])), None, None]))
         / 3f64
 }