From c8c05d29419822aff3554af788e910ec69267406 Mon Sep 17 00:00:00 2001
From: Dennis Kobert <dennis@kobert.dev>
Date: Wed, 2 Apr 2025 10:20:43 +0200
Subject: Implement energy model for e cores

---
 src/model.rs | 16 ++--------------
 1 file changed, 2 insertions(+), 14 deletions(-)

(limited to 'src/model.rs')

diff --git a/src/model.rs b/src/model.rs
index fbb6e0b..6dbde1f 100644
--- a/src/model.rs
+++ b/src/model.rs
@@ -50,22 +50,10 @@ impl<B: Backend> Net<B> {
 }
 
 /// Load the p core model from the file in your source code (not in build.rs or script).
-pub fn load_model_p() -> Net<ArrayBackend> {
+pub fn load_model(path: &str) -> Net<ArrayBackend> {
     let device = Default::default();
     let record: NetRecord<ArrayBackend> = PyTorchFileRecorder::<FullPrecisionSettings>::default()
-        .load("./perf.pt".into(), &device)
-        .expect("Failed to decode state");
-
-    Net::<ArrayBackend>::init(&device).load_record(record)
-}
-
-/// Load the e core model from the file in your source code (not in build.rs or script).
-pub fn load_model_e() -> Net<ArrayBackend> {
-    //TODO: load e model
-    println!("Falling back to p model");
-    let device = Default::default();
-    let record: NetRecord<ArrayBackend> = PyTorchFileRecorder::<FullPrecisionSettings>::default()
-        .load("./perf.pt".into(), &device)
+        .load(path.into(), &device)
         .expect("Failed to decode state");
 
     Net::<ArrayBackend>::init(&device).load_record(record)
-- 
cgit v1.3.1