search

WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
23.33k stars 8.02k forks source link

Faster RCNN中添加validation loss的问题 #597

Closed chenyuuu0705 closed 2 years ago

chenyuuu0705 commented 2 years ago

请问，我如果想在faster rcnn 中添加训练过程中 validation的loss的话，像图中这样的写法是正确的吗？ image

WZMIAOMIAO commented 2 years ago

不行，这样做验证集也会参与训练

chenyuuu0705 commented 2 years ago

@WZMIAOMIAO 你好，我在vgg的代码中想要计算train accuracy,然后进行了修改，如图所示，请问这样可以吗？ image

WZMIAOMIAO commented 2 years ago

你直接参考我提供的项目代码改吧。 https://github.com/WZMIAOMIAO/deep-learning-for-image-processing/blob/309520fb600b24eafebed3834fcee52ef1ba4e6e/pytorch_classification/Test11_efficientnetV2/train.py#L97-L109

chenyuuu0705 commented 2 years ago

@WZMIAOMIAO 非常感谢 ：）