From e8cf619005860e758800c8a82f0ddd7ad60b94b7 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 24 May 2019 00:01:37 -0700 Subject: [PATCH] Update gluon_resnet.py Update header comment --- models/gluon_resnet.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/models/gluon_resnet.py b/models/gluon_resnet.py index 9918b3e5..5b6ad78c 100644 --- a/models/gluon_resnet.py +++ b/models/gluon_resnet.py @@ -1,8 +1,7 @@ -"""Pytorch ResNet implementation w/ tweaks -This file is a copy of https://github.com/pytorch/vision 'resnet.py' (BSD-3-Clause) with -additional dropout and dynamic global avg/max pool. - -ResNext additions added by Ross Wightman +"""Pytorch impl of MxNet Gluon ResNet/(SE)ResNeXt variants +This file evolved from https://github.com/pytorch/vision 'resnet.py' with (SE)-ResNeXt additions +and ports of Gluon variations (https://github.com/dmlc/gluon-cv/blob/master/gluoncv/model_zoo/resnet.py) +by Ross Wightman """ import torch import torch.nn as nn