From 6478bcd02c8af1b238d6ecb7c2eba247e91b1246 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sun, 26 Sep 2021 14:54:17 -0700 Subject: [PATCH] Fix regnetz_d conv layer name, use inception mean/std --- timm/models/byobnet.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/timm/models/byobnet.py b/timm/models/byobnet.py index 50ad1e88..d00aeb32 100644 --- a/timm/models/byobnet.py +++ b/timm/models/byobnet.py @@ -141,13 +141,16 @@ default_cfgs = { # experimental models 'regnetz_b': _cfg( url='', + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), 'regnetz_c': _cfg( url='', + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), 'regnetz_d': _cfg( url='', - input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), + mean=(0.5, 0.5, 0.5), std=(0.5, 0.5, 0.5), + first_conv='stem.conv1.conv', input_size=(3, 256, 256), pool_size=(8, 8), interpolation='bicubic'), }