2021
07-17
07-17
Pytorch中的数据集划分&正则化方法
1.训练集&验证集&测试集训练集:训练数据验证集:验证不同算法(比如利用网格搜索对超参数进行调整等),检验哪种更有效测试集:正确评估分类器的性能正常流程:验证集会记录每个时间戳的参数,在加载test数据前会加载那个最好的参数,再来评估。比方说训练完6000个epoch后,发现在第3520个epoch的validation表现最好,测试时会加载第3520个epoch的参数。importtorchimporttorch.nnasnnimporttorch.nn.functionalasFimpo...
继续阅读 >