diff --git a/tutorials/05 - Deep Residual Network/main-gpu.py b/tutorials/05 - Deep Residual Network/main-gpu.py index 14cc3f7..2da4b0f 100644 --- a/tutorials/05 - Deep Residual Network/main-gpu.py +++ b/tutorials/05 - Deep Residual Network/main-gpu.py @@ -1,5 +1,7 @@ -# Implementation of https://arxiv.org/pdf/1512.03385.pdf -# See section 4.2 for model architecture on CIFAR-10 +# Implementation of https://arxiv.org/pdf/1512.03385.pdf/ +# See section 4.2 for model architecture on CIFAR-10. +# Some part of the code was referenced below. +# https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py import torch import torch.nn as nn import torchvision.datasets as dsets diff --git a/tutorials/05 - Deep Residual Network/main.py b/tutorials/05 - Deep Residual Network/main.py index aad0da2..685d964 100644 --- a/tutorials/05 - Deep Residual Network/main.py +++ b/tutorials/05 - Deep Residual Network/main.py @@ -1,5 +1,7 @@ -# Implementation of https://arxiv.org/pdf/1512.03385.pdf -# See section 4.2 for model architecture on CIFAR-10 +# Implementation of https://arxiv.org/pdf/1512.03385.pdf. +# See section 4.2 for model architecture on CIFAR-10. +# Some part of the code was referenced below. +# https://github.com/pytorch/vision/blob/master/torchvision/models/resnet.py import torch import torch.nn as nn import torchvision.datasets as dsets