Avoid a 2nd forward pass during validation step #857
Labels
API
This tag is used for small improvements to the readability and usability of the python API.
Feature Request
New feature or request
Performance
Question on model performance and accuracy
Deepforest primary connects to Torchvision models. Torchvision models have a 'train' and 'eval' mode, which are uncomfortably named, because it also effects what is outputted. In train mode, the loss is returned, in eval mode, the predictions are returned.
https://discuss.pytorch.org/t/how-to-get-losses-and-predictions-at-the-same-time/167223
from https://pytorch.org/vision/main/models/generated/torchvision.models.detection.retinanet_resnet50_fpn.html#torchvision.models.detection.retinanet_resnet50_fpn
The line of code is here.
DeepForest/src/deepforest/main.py
Line 677 in e5f3301
The text was updated successfully, but these errors were encountered: