cargo fmt

open-spaced-repetition · Aug 25, 2023 · 6974454 · 6974454
1 parent 06a9863
commit 6974454
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/training.rs b/src/training.rs
@@ -35,8 +35,7 @@ impl<B: Backend<FloatElem = f32>> Model<B> {
  delta_ts.clone().unsqueeze::<2>().transpose(),
  stability.clone(),
  );
- let logits =
- Tensor::cat(vec![-retention.clone() + 1, retention.clone()], 1);
+ let logits = Tensor::cat(vec![-retention.clone() + 1, retention.clone()], 1);
  info!("stability: {}", &stability);
  info!(
  "delta_ts: {}",