COVID-CT新冠肺炎检测(DenseNet网络)
train()函数是训练过程的核心。在这个函数里,完成了一个epoch的数据的训练。过程是从train_loader中去取一个batch_size的图像和标签,前向传播,计算交叉熵loss,反向传播,完成一次迭代。过完全部数据,跳出函数。device=torch.device("cpu")#使用cpu训练,或者GPU#device=torch.device("GPU")#使用cpu训练,或者GPUmodel.train()#训练模式bs=32'''model。...
原创
2022-07-31 15:15:13 ·
2313 阅读 ·
2 评论