diff --git a/timm/models/resnet.py b/timm/models/resnet.py index f417ef89..6a128bfe 100644 --- a/timm/models/resnet.py +++ b/timm/models/resnet.py @@ -1,6 +1,8 @@ """PyTorch ResNet + This started as a copy of https://github.com/pytorch/vision 'resnet.py' (BSD-3-Clause) with additional dropout and dynamic global avg/max pool. + ResNeXt, SE-ResNeXt, SENet, and MXNet Gluon stem/downsample variants, tiered stems added by Ross Wightman Copyright 2020 Ross Wightman """