标签： deep-learning computer-vision conv-neural-network pytorch
如何确定CNN中的层数和最终模型以提高预测的准确性。 我正在对图像进行分类，目前使用简单的模型可以获得65％的准确度，我应该如何增强它以达到最大的准确度。 （火炬）
答案 0 :(得分：0)
我会说三件事。
1）检查torchvision.models link，您可以找到经过预训练的出色模型，如果您将图层设置为不需要渐变，而只是修改最终图层以具有正确的类数，则会为您提供出色的性能 >
2）加载图像link时可以进行变换，这可以为您
3）使用最后一层的数量，使用不同的优化程序并尝试使用调度程序link（这将在训练过程中调整学习率，以使其更合适）
希望它会有所帮助：）