Skip to content

人
人工智能系统实战第三期
Commit 7e13af9b by 前钰
Options

Update train.py

parent 9ae0c471
Showing with 2 additions and 2 deletions
人工智能系统实战第三期/实战代码/深度学习项目实战/基于transformer的花朵识别/ViTFlowerClassification/train.py
import gc import gc
...@@ -9,7 +9,7 @@ from tqdm import tqdm ...@@ -9,7 +9,7 @@ from tqdm import tqdm
import torch import torch
import torch.optim as optim import torch.optim as optim
import torch.optim.lr_scheduler as lr_scheduler import torch.optim.lr_scheduler as lr_scheduler
from torch.utils.tensorboard import SummaryWriter from tensorboardX import SummaryWriter
from dataloader import get_data_loader from dataloader import get_data_loader
from utils import remove_dir_and_create_dir, create_model, model_parallel, set_seed from utils import remove_dir_and_create_dir, create_model, model_parallel, set_seed
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment