From 4c4138fa33dcd7345fb86d263b4c180126ff029a Mon Sep 17 00:00:00 2001
From: rvandeghen <37592623+rvandeghen@users.noreply.github.com>
Date: Tue, 10 May 2022 10:39:13 +0200
Subject: [PATCH] Change pad value to int

Solve the problem of (#5913)
---
 references/detection/presets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/references/detection/presets.py b/references/detection/presets.py
index 779f3f218ca..6682385c94b 100644
--- a/references/detection/presets.py
+++ b/references/detection/presets.py
@@ -3,7 +3,7 @@
 
 
 class DetectionPresetTrain:
-    def __init__(self, *, data_augmentation, hflip_prob=0.5, mean=(123.0, 117.0, 104.0)):
+    def __init__(self, *, data_augmentation, hflip_prob=0.5, mean=(123, 117, 104)):
         if data_augmentation == "hflip":
             self.transforms = T.Compose(
                 [