weiyuqing/ECG_by_Diffenert_CNN_Test
1
0 Fork 0
mirror of https://github.com/This-weiyuqing/ECG_by_Diffenert_CNN_Test.git synced 2026-10-18 03:13:39 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update _main_pytorch_threeLayerCNN.py

Browse Source
This commit is contained in:
weiyuqing committed 2023-03-21 20:53:06 +08:00
1 parent 23815440f7
commit ab85e79e3a
1 file changed
+3 -3
Download Patch File Download Diff File Expand all files Collapse all files
Number-Of-CNN-Layers/_main_pytorch_threeLayerCNN.py
+3 -3
View File
@@ -11,15 +11,15 @@ from torch.utils.tensorboard import SummaryWriter
from sklearn.metrics import accuracy_score from sklearn.metrics import accuracy_score
from tqdm import tqdm 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 root path
project_path = "./" project_path = "../Number-Of-CNN-Layers/Three/"
# define log directory # define log directory
# must be a subdirectory of the directory specified when starting the web application # 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 # 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") 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 # the device to use
device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")