|
1 | 1 | """
|
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 | +
|
7 | 9 |
|
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. |
11 | 10 |
|
12 | 11 | """
|
13 | 12 |
|
14 | 13 |
|
15 | 14 | ######################################################################
|
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 | + |
18 | 27 |
|
19 | 28 | import torch.optim as optim
|
20 | 29 | import albumentations as A
|
|
30 | 39 |
|
31 | 40 |
|
32 | 41 | ######################################################################
|
33 |
| -# I recommend using GPU for this tutorial. |
| 42 | +# We recommend using GPU for this tutorial. |
34 | 43 | #
|
35 | 44 |
|
36 | 45 | device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
|
126 | 135 |
|
127 | 136 | ######################################################################
|
128 | 137 | # | 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 . |
130 | 139 | # | You can modify the batch size according to your preference.
|
131 | 140 | #
|
132 | 141 |
|
|
0 commit comments