From 2de54d174a5c76791b8ff4991ae7a1662d180c4b Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Sat, 30 Jan 2021 18:02:33 -0800 Subject: [PATCH] Fix pool size defs for NFNet models, add a comment. --- timm/models/nfnet.py | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/timm/models/nfnet.py b/timm/models/nfnet.py index f54bb84a..4e5253a9 100644 --- a/timm/models/nfnet.py +++ b/timm/models/nfnet.py @@ -3,6 +3,9 @@ Paper: `Characterizing signal propagation to close the performance gap in unnormalized ResNets` - https://arxiv.org/abs/2101.08692 +NOTE: These models are a work in progress, no pretrained weights yet but I'm currently training some. +Details may change, especially once the paper authors release their official models. + Hacked together by / copyright Ross Wightman, 2021. """ import math @@ -34,11 +37,11 @@ def _dcfg(url='', **kwargs): # FIXME finish default_cfgs = { 'nf_regnet_b0': _dcfg(url=''), - 'nf_regnet_b1': _dcfg(url='', input_size=(3, 240, 240)), - 'nf_regnet_b2': _dcfg(url='', input_size=(3, 256, 256)), - 'nf_regnet_b3': _dcfg(url='', input_size=(3, 272, 272)), - 'nf_regnet_b4': _dcfg(url='', input_size=(3, 320, 320)), - 'nf_regnet_b5': _dcfg(url='', input_size=(3, 384, 384)), + 'nf_regnet_b1': _dcfg(url='', input_size=(3, 240, 240), pool_size=(8, 8)), + 'nf_regnet_b2': _dcfg(url='', input_size=(3, 256, 256), pool_size=(8, 8)), + 'nf_regnet_b3': _dcfg(url='', input_size=(3, 272, 272), pool_size=(9, 9)), + 'nf_regnet_b4': _dcfg(url='', input_size=(3, 320, 320), pool_size=(10, 10)), + 'nf_regnet_b5': _dcfg(url='', input_size=(3, 384, 384), pool_size=(12, 12)), 'nf_resnet26': _dcfg(url='', first_conv='stem.conv'), 'nf_resnet50': _dcfg(url='', first_conv='stem.conv'),