diff --git a/model.py b/Model.py similarity index 100% rename from model.py rename to Model.py diff --git a/train.py b/Train.py similarity index 99% rename from train.py rename to Train.py index cf67ef5..6c87c69 100644 --- 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"""