reference added"

Author: yunjey

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

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