Skip to content

Commit e8cf619

Browse files
authored
Update gluon_resnet.py
Update header comment
1 parent 7419e98 commit e8cf619

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

models/gluon_resnet.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,8 +1,7 @@
1-
"""Pytorch ResNet implementation w/ tweaks
2-
This file is a copy of https://github.com/pytorch/vision 'resnet.py' (BSD-3-Clause) with
3-
additional dropout and dynamic global avg/max pool.
4-
5-
ResNext additions added by Ross Wightman
1+
"""Pytorch impl of MxNet Gluon ResNet/(SE)ResNeXt variants
2+
This file evolved from https://github.com/pytorch/vision 'resnet.py' with (SE)-ResNeXt additions
3+
and ports of Gluon variations (https://github.com/dmlc/gluon-cv/blob/master/gluoncv/model_zoo/resnet.py)
4+
by Ross Wightman
65
"""
76
import torch
87
import torch.nn as nn

0 commit comments

Comments
 (0)