Skip to content

Commit c401bcb

Browse files
committed
modify : tutorial
Follow Pytorch Tutorial convetion Modify I -> We
1 parent 6e92650 commit c401bcb

File tree

1 file changed

+21
-12
lines changed

1 file changed

+21
-12
lines changed

beginner_source/Pretraining_Vgg_from_scratch.py

Lines changed: 21 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,29 @@
11
"""
2-
In this tutorial, we will embark on an exciting journey to build and
3-
train a VGG network from scratch using Python and popular deep learning
4-
libraries such as PyTorch. We will dive into the details of the VGG
5-
architecture, understanding its components and the rationale behind its
6-
design.
2+
Pretraining VGG from scratch
3+
============================
4+
5+
6+
**Author:** `WoongJoon Choi <https://github.com/woongjoonchoi>`_
7+
8+
79
8-
Our tutorial is designed for both beginners who are new to deep learning
9-
and seasoned practitioners looking to deepen their understanding of CNN
10-
architectures.
1110
1211
"""
1312

1413

1514
######################################################################
16-
# Author : `WoongJoon Choi <https://github.com/woongjoonchoi>`__
17-
#
15+
# In this tutorial, we will embark on an exciting journey to build and
16+
# train a VGG network from scratch using Python and popular deep learning
17+
# libraries such as PyTorch. We will dive into the details of the VGG
18+
# architecture, understanding its components and the rationale behind its
19+
# design.
20+
#
21+
# Our tutorial is designed for both beginners who are new to deep learning
22+
# and seasoned practitioners looking to deepen their understanding of CNN
23+
# architectures.
24+
25+
26+
1827

1928
import torch.optim as optim
2029
import albumentations as A
@@ -30,7 +39,7 @@
3039

3140

3241
######################################################################
33-
# I recommend using GPU for this tutorial.
42+
# We recommend using GPU for this tutorial.
3443
#
3544

3645
device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -126,7 +135,7 @@
126135

127136
######################################################################
128137
# | If your GPU memory is 24GB ,The maximum batch size is 128. But, if you
129-
# use Colab , I recommend using 32 .
138+
# use Colab , We recommend using 32 .
130139
# | You can modify the batch size according to your preference.
131140
#
132141

0 commit comments

Comments
 (0)