BooTurbo

摘要： 神经网络每训练1个epoch 会紧接着 evaluate一下，PyTorch提供了model.eval( )和torch.no_grad( )，验证时会使用 model.eval( ) 开启评估模式， 在训练阶段，网络的Dropout 层和BatchNorm 层处于开启的模式； 在验证阶段，代码中加 阅读全文

摘要： 在训练神经网络时，用到的数据量可能很大，训练周期较长，如果半途中断了训练，下次从头训练就会很费时间，这时我们就想断点续训。 一、神经网络模型的保存，基本两种方式：1. 保存完整模型model， torch.save(model, save_path) 2. 只保存模型的参数， torch.save( 阅读全文