Skip to content

Commit f0780f0

Browse files
committed
fix rst syntax in ddp_pipeline.py
1 parent b908557 commit f0780f0

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

โ€Žadvanced_source/ddp_pipeline.pyโ€Ž

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,6 @@
11
"""
22
๋ถ„์‚ฐ ๋ฐ์ดํ„ฐ ๋ณ‘๋ ฌ ์ฒ˜๋ฆฌ์™€ ํŒŒ์ดํ”„๋ผ์ธ ๋ณ‘๋ ฌํ™”๋ฅผ ์‚ฌ์šฉํ•œ ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋ธ ํ•™์Šต
3-
====================================================================================
3+
=======================================================================
44
55
**Author**: `Pritam Damania <https://github.com/pritamdamania87>`_
66
**๋ฒˆ์—ญ**: `๋ฐฑ์„ ํฌ <https://github.com/spongebob03>`_
@@ -22,14 +22,14 @@
2222

2323
######################################################################
2424
# ๋ชจ๋ธ ์ •์˜ํ•˜๊ธฐ
25-
# ----------------
25+
# -------------
2626
#
2727

2828
######################################################################
2929
# ``PositionalEncoding`` ๋ชจ๋“ˆ์€ ์‹œํ€€์Šค์—์„œ ํ† ํฐ์˜ ์ƒ๋Œ€์ , ์ ˆ๋Œ€ ์œ„์น˜์— ๋Œ€ํ•œ
3030
# ์ผ๋ถ€ ์ •๋ณด๋ฅผ ์ฃผ์ž…ํ•ฉ๋‹ˆ๋‹ค.
3131
# ์œ„์น˜ ์ธ์ฝ”๋”ฉ์€ ์ž„๋ฒ ๋”ฉ๊ณผ ๊ฐ™์€ ์ฐจ์›์„ ๊ฐ€์ง€๋ฏ€๋กœ
32-
# ๋‘˜์„ ํ•ฉ์น  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ, ์ฃผํŒŒ์ˆ˜๊ฐ€ ๋‹ค๋ฅธ ``sine``๊ณผ ``cosine`` ๊ธฐ๋Šฅ์„
32+
# ๋‘˜์„ ํ•ฉ์น  ์ˆ˜ ์žˆ์Šต๋‹ˆ๋‹ค. ์—ฌ๊ธฐ์„œ, ์ฃผํŒŒ์ˆ˜๊ฐ€ ๋‹ค๋ฅธ ``sine`` ๊ณผ ``cosine`` ๊ธฐ๋Šฅ์„
3333
# ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.
3434

3535
import sys
@@ -73,7 +73,7 @@ def forward(self, x):
7373
# `nn.TransformerEncoder <https://pytorch.org/docs/stable/generated/torch.nn.TransformerEncoder.html>`__ ๊ณ„์ธต(layer)์— ํฌํ•จ๋ฉ๋‹ˆ๋‹ค.
7474
# `nn.TransformerEncoder <https://pytorch.org/docs/stable/generated/torch.nn.TransformerEncoder.html>`__ ๋Š”
7575
# `nn.TransformerEncoderLayer <https://pytorch.org/docs/stable/generated/torch.nn.TransformerEncoderLayer.html>`__ ์˜ ``nlayers`` ๋กœ ๊ตฌ์„ฑ๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.
76-
# ๊ฒฐ๊ณผ์ ์œผ๋กœ, ์šฐ๋ฆฌ๋Š” ``nn.TransformerEncoder`` ์— ์ค‘์ ์„ ๋‘๊ณ  ์žˆ์œผ๋ฉฐ
76+
# ๊ฒฐ๊ณผ์ ์œผ๋กœ, ``nn.TransformerEncoder`` ์— ์ค‘์ ์„ ๋‘๊ณ  ์žˆ์œผ๋ฉฐ
7777
# ``nn.TransformerEncoderLayer`` ์˜ ์ ˆ๋ฐ˜์€ ํ•œ GPU์— ๋‘๊ณ 
7878
# ๋‚˜๋จธ์ง€ ์ ˆ๋ฐ˜์€ ๋‹ค๋ฅธ GPU์— ์žˆ๋„๋ก ๋ชจ๋ธ์„ ๋ถ„ํ• ํ•ฉ๋‹ˆ๋‹ค. ์ด๋ฅผ ์œ„ํ•ด์„œ ``Encoder`` ์™€
7979
# ``Decoder`` ์„น์…˜์„ ๋ถ„๋ฆฌ๋œ ๋ชจ๋“ˆ๋กœ ๋นผ๋‚ธ ๋‹ค์Œ, ์›๋ณธ ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋“ˆ์„
@@ -122,20 +122,20 @@ def forward(self, inp):
122122

123123
######################################################################
124124
# ํ•™์Šต์„ ์œ„ํ•œ ๋‹ค์ค‘ ํ”„๋กœ์„ธ์Šค ์‹œ์ž‘
125-
# -------------------------------------
125+
# ------------------------------
126126
#
127127

128128

129129
######################################################################
130130
# ๊ฐ์ž ๋‘ ๊ฐœ์˜ GPU์—์„œ ์ž์ฒด ํŒŒ์ดํ”„๋ผ์ธ์„ ๊ตฌ๋™ํ•˜๋Š” ๋‘ ๊ฐ€์ง€ ํ”„๋กœ์„ธ์Šค๋ฅผ ์‹œ์ž‘ํ•ฉ๋‹ˆ๋‹ค.
131-
# ``run_worker``๋Š” ๊ฐ ํ”„๋กœ์„ธ์Šค์— ์‹คํ–‰๋ฉ๋‹ˆ๋‹ค.
131+
# ``run_worker`` ๋Š” ๊ฐ ํ”„๋กœ์„ธ์Šค์— ์‹คํ–‰๋ฉ๋‹ˆ๋‹ค.
132132

133133
def run_worker(rank, world_size):
134134

135135

136136
######################################################################
137137
# ๋ฐ์ดํ„ฐ ๋กœ๋“œํ•˜๊ณ  ๋ฐฐ์น˜ ๋งŒ๋“ค๊ธฐ
138-
# -------------------
138+
# ---------------------------
139139
#
140140

141141

@@ -210,7 +210,7 @@ def batchify(data, bsz, rank, world_size, is_train=False):
210210

211211
######################################################################
212212
# ์ž…๋ ฅ๊ณผ ํƒ€๊ฒŸ ์‹œํ€€์Šค๋ฅผ ์ƒ์„ฑํ•˜๊ธฐ ์œ„ํ•œ ํ•จ์ˆ˜๋“ค
213-
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
213+
# ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
214214
#
215215

216216

@@ -239,15 +239,15 @@ def get_batch(source, i):
239239

240240
######################################################################
241241
# ๋ชจ๋ธ ๊ทœ๋ชจ์™€ ํŒŒ์ดํ”„ ์ดˆ๊ธฐํ™”
242-
# -----------------------------------
242+
# -------------------------
243243
#
244244

245245

246246
######################################################################
247247
# ํŒŒ์ดํ”„๋ผ์ธ ๋ณ‘๋ ฌํ™”๋ฅผ ํ™œ์šฉํ•œ ๋Œ€ํ˜• ํŠธ๋žœ์Šคํฌ๋จธ ๋ชจ๋ธ ํ•™์Šต์„ ์ฆ๋ช…ํ•˜๊ธฐ ์œ„ํ•ด,
248248
# ํŠธ๋žœ์Šคํฌ๋จธ ๊ณ„์ธต ๊ทœ๋ชจ๋ฅผ ์ ์ ˆํžˆ ํ™•์žฅ์‹œํ‚ต๋‹ˆ๋‹ค. We use an embedding
249249
# 4096์ฐจ์›์˜ ์ž„๋ฒ ๋”ฉ ๋ฒกํ„ฐ, 4096์˜ ์€๋‹‰ ์‚ฌ์ด์ฆˆ, 16๊ฐœ์˜ ์–ดํ…์…˜ ํ—ค๋“œ(attention head)์™€ ์ด 8 ๊ฐœ์˜
250-
# ํŠธ๋žœ์Šคํฌ๋จธ ๊ณ„์ธต (``nn.TransformerEncoderLayer``)๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค. ์ด๋Š” ์ตœ๋Œ€
250+
# ํŠธ๋žœ์Šคํฌ๋จธ ๊ณ„์ธต (``nn.TransformerEncoderLayer``)๋ฅผ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค. ์ด๋Š” ์ตœ๋Œ€
251251
# **~1 ์–ต** ๊ฐœ์˜ ํŒŒ๋ผ๋ฏธํ„ฐ๋ฅผ ๊ฐ–๋Š” ๋ชจ๋ธ์„ ์ƒ์„ฑํ•ฉ๋‹ˆ๋‹ค.
252252
#
253253
# `RPC ํ”„๋ ˆ์ž„์›Œํฌ <https://pytorch.org/docs/stable/rpc.html>`__ ๋ฅผ ์ดˆ๊ธฐํ™”ํ•ด์•ผ ํ•ฉ๋‹ˆ๋‹ค.
@@ -434,7 +434,7 @@ def evaluate(eval_model, data_source):
434434

435435
######################################################################
436436
# ํ‰๊ฐ€ ๋ฐ์ดํ„ฐ์…‹์œผ๋กœ ๋ชจ๋ธ ํ‰๊ฐ€ํ•˜๊ธฐ
437-
# -------------------------------------
437+
# -------------------------------
438438
#
439439
# ํ‰๊ฐ€ ๋ฐ์ดํ„ฐ์…‹์—์„œ์˜ ๊ฒฐ๊ณผ๋ฅผ ํ™•์ธํ•˜๊ธฐ ์œ„ํ•ด ์ตœ๊ณ ์˜ ๋ชจ๋ธ์„ ์ ์šฉํ•ฉ๋‹ˆ๋‹ค.
440440

0 commit comments

Comments
ย (0)