Update gluon_resnet.py

Update header comment
pull/2/head
Ross Wightman 6 years ago committed by GitHub
parent 7419e9835f
commit e8cf619005
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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

Loading…
Cancel
Save