yukun-hh/trash-division
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

模型规范重命名

Browse source
This commit is contained in:
yukun-hh 2026-04-12 14:39:58 +08:00
parent 39e8781bdd
commit af1bed54ea
2 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

0
model.py → Model.py
View file

2
train.py → Train.py
View file

@ -11,7 +11,7 @@ import torch.nn as nn
import torch.optim as optim import torch.optim as optim
from tqdm import tqdm # 进度条，可选 from tqdm import tqdm # 进度条，可选
import matplotlib.pyplot as plt import matplotlib.pyplot as plt
from model import Net from Model import Net
def train_one_epoch(model, train_loader, criterion, optimizer, device, epoch): def train_one_epoch(model, train_loader, criterion, optimizer, device, epoch):
"""训练一个epoch""" """训练一个epoch"""