Skip to content

Commit 97e2d52

Browse files
authored
Merge pull request #1861 from Trusted-AI/dev_1.12.1
Update to ART 1.12.1
2 parents 89bf92f + 0235218 commit 97e2d52

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

art/estimators/object_detection/pytorch_yolo.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -253,6 +253,8 @@ def _get_losses(
253253
import torch # lgtm [py/repeated-import]
254254

255255
self._model.train()
256+
self.set_batchnorm(train=False)
257+
self.set_dropout(train=False)
256258

257259
# Apply preprocessing
258260
if self.all_framework_preprocessing:

0 commit comments

Comments
 (0)