Cleanup

tracel-ai · Apr 5, 2023 · b30278a · b30278a
1 parent b6f6e2b
commit b30278a
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 13 deletions.
diff --git a/burn-derive/src/lib.rs b/burn-derive/src/lib.rs
@@ -18,8 +18,6 @@ pub fn module_derive(input: TokenStream) -> TokenStream {
 #[proc_macro_derive(Record)]
 pub fn record_derive(input: TokenStream) -> TokenStream {
     let input = syn::parse(input).unwrap();
-
-    // panic!("{}", gen);
     record_derive_impl(&input)
 }
 

diff --git a/burn-tensor/src/tensor/data.rs b/burn-tensor/src/tensor/data.rs
@@ -223,7 +223,6 @@ impl<E: Into<f64> + Clone + core::fmt::Debug + PartialEq, const D: usize> Data<E
             let b = round(pow(10.0_f64, precision as f64) * b);
 
             if a != b {
-                println!("{a:?} != {b:?}");
                 eq = false;
             }
         }

diff --git a/burn-train/src/checkpoint/file.rs b/burn-train/src/checkpoint/file.rs
@@ -54,21 +54,15 @@ where
             return Ok(());
         }
 
-        let file_path_old_checkpoint = self.path_for_epoch(epoch - self.num_keep);
         let file_to_remove = format!(
             "{}.{}",
-            file_path_old_checkpoint,
+            self.path_for_epoch(epoch - self.num_keep),
             <S::Recorder as FileRecorder>::file_extension()
         );
 
-        match std::fs::remove_file(file_to_remove) {
-            Ok(_) => log::info!("Removed checkpoint {}", file_path_old_checkpoint),
-            Err(err) => {
-                match err.kind() {
-                    std::io::ErrorKind::NotFound => (), // Ignoring missing old checkpoints,
-                    _ => return Err(CheckpointerError::IOError(err)),
-                }
-            }
+        if std::path::Path::new(&file_to_remove).exists() {
+            log::info!("Removing checkpoint {}", file_to_remove);
+            std::fs::remove_file(file_to_remove).map_err(CheckpointerError::IOError)?;
         }
 
         Ok(())