updates

2026-05-23 23:12:27 -05:00
parent 764252b0c4
commit 7d5c0cf7eb
3 changed files with 16 additions and 14 deletions
@@ -313,15 +313,18 @@ impl<B: Backend> Batcher<B, TrainingSample, ChessBatch<B>> for ChessBatcher {
                        *p /= sum;
                    }
                }
-                Tensor::<B, 2>::from_floats(TensorData::new(policy, [4672]), device).unsqueeze()
+                Tensor::<B, 2>::from_floats(TensorData::new(policy, [1, 4672]), device).unsqueeze()
            })
            .collect::<Vec<_>>();

        let value_target_tensors = items
            .iter()
            .map(|item| {
-                Tensor::<B, 2>::from_floats(TensorData::new(vec![item.value_target], [1]), device)
-                    .reshape([1, 1])
+                Tensor::<B, 2>::from_floats(
+                    TensorData::new(vec![item.value_target], [1, 1]),
+                    device,
+                )
+                .reshape([1, 1])
            })
            .collect::<Vec<_>>();