Adding disablement of batch norm

Signed-off-by: Ben <[email protected]>

Author: Benjamin-Etheredge

art/estimators/object_detection/pytorch_object_detector.py

@@ -310,6 +310,7 @@ def _get_losses(
         self._model.train()
 
         self.set_dropout(False)
+        self.set_batchnorm(False)
         self.set_multihead_attention(False)
 
         # Apply preprocessing and convert to tensors