From ab85e79e3ac5e97df0173ca5d584f3a711705497 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E9=AD=8F=E9=81=87=E5=8D=BF?= <358614401@qq.com>
Date: Tue, 21 Mar 2023 20:53:06 +0800
Subject: [PATCH] Update _main_pytorch_threeLayerCNN.py

---
 Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py b/Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py
index 5afe3ad..839e16e 100644
--- a/Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py
+++ b/Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py
@@ -11,15 +11,15 @@ from torch.utils.tensorboard import SummaryWriter
 from sklearn.metrics import accuracy_score
 from tqdm import tqdm
 
-from utils import load_data, plot_history_torch, plot_heat_map
+from utils_threeLayerCNN import loadData, plot_history_torch, plot_heat_map
 
 # project root path
-project_path = "./"
+project_path = "../Number-Of-CNN-Layers/Three/"
 # define log directory
 # must be a subdirectory of the directory specified when starting the web application
 # it is recommended to use the date time as the subdirectory name
 log_dir = project_path + "logs/" + datetime.datetime.now().strftime("%Y%m%d-%H%M%S")
-model_path = project_path + "ecg_model.pt"
+model_path = project_path + "model/ecg_model_three_layer.pt"
 
 # the device to use
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")