2022年7月1日
torch训练和推理的模板
摘要： def train(epoch): model.train() train_loss = 0 for data, label in train_loader: data, label = data.cuda(), label.cuda() # 将数据放入显卡 optimizer.zero_grad( 阅读全文
posted @ 2022-07-01 20:12 乌蝇哥 阅读(376) 评论(0) 推荐(0) 编辑