tools.train_model

tools.train_model(model, loader, criterion, optimizer, scheduler, num_epochs=25, device='cpu')

模型训练所支持的函数

参数

• model – 待训练的模型

• criterion – 优化准则 (loss)

• optimizer – 用于训练的 Optimizer

• scheduler – torch.optim.lr_scheduler 中的 LR 调度器对象

• num_epochs – epochs 数量

• device – 用来进行训练的设备。必须是 ‘cpu’ 或 ‘cuda’

上一页

工具

下一页

生态系统