模型规范重命名

2026-04-12 14:39:58 +08:00 · 2026-04-12 14:39:58 +08:00 · af1bed54ea
commit af1bed54ea
parent 39e8781bdd
2 changed files with 1 additions and 1 deletions
--- a/Model.py
+++ b/Model.py
--- a/Train.py
+++ b/Train.py
@ -11,7 +11,7 @@ import torch.nn as nn
 import torch.optim as optim
 from tqdm import tqdm  # 进度条，可选
 import matplotlib.pyplot as plt
-from model import Net
+from Model import Net

 def train_one_epoch(model, train_loader, criterion, optimizer, device, epoch):
    """训练一个epoch"""