Skip to content

Commit 1726ae9

Browse files
authored
Merge branch 'main' into issues/1269
2 parents 060dc6a + 66eaf6a commit 1726ae9

File tree

9 files changed

+462
-11
lines changed

9 files changed

+462
-11
lines changed

.pyspelling.yml

+2-2
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@ matrix:
1919
- open: '\.\.\s+(figure|literalinclude|math|image|grid)::'
2020
close: '\n'
2121
# Exclude roles:
22-
- open: ':(?:(class|py:mod|mod|func)):`'
22+
- open: ':(?:(class|py:mod|mod|func|meth|obj)):`'
2323
content: '[^`]*'
2424
close: '`'
2525
# Exclude reStructuredText hyperlinks
@@ -70,7 +70,7 @@ matrix:
7070
- open: ':figure:.*'
7171
close: '\n'
7272
# Ignore reStructuredText roles
73-
- open: ':(?:(class|file|func|math|ref|octicon)):`'
73+
- open: ':(?:(class|file|func|math|ref|octicon|meth|obj)):`'
7474
content: '[^`]*'
7575
close: '`'
7676
- open: ':width:'

_static/img/rollout_recurrent.png

338 KB
Loading

beginner_source/introyt/tensorboardyt_tutorial.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -214,13 +214,14 @@ def forward(self, x):
214214
# Check against the validation set
215215
running_vloss = 0.0
216216

217-
net.train(False) # Don't need to track gradents for validation
217+
# In evaluation mode some model specific operations can be omitted eg. dropout layer
218+
net.train(False) # Switching to evaluation mode, eg. turning off regularisation
218219
for j, vdata in enumerate(validation_loader, 0):
219220
vinputs, vlabels = vdata
220221
voutputs = net(vinputs)
221222
vloss = criterion(voutputs, vlabels)
222223
running_vloss += vloss.item()
223-
net.train(True) # Turn gradients back on for training
224+
net.train(True) # Switching back to training mode, eg. turning on regularisation
224225

225226
avg_loss = running_loss / 1000
226227
avg_vloss = running_vloss / len(validation_loader)

en-wordlist.txt

+1
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ Colab
6262
Conv
6363
ConvNet
6464
ConvNets
65+
customizable
6566
DCGAN
6667
DCGANs
6768
DDP

index.rst

+7-2
Original file line numberDiff line numberDiff line change
@@ -312,15 +312,20 @@ What's new in PyTorch tutorials?
312312
:link: intermediate/mario_rl_tutorial.html
313313
:tags: Reinforcement-Learning
314314

315+
.. customcarditem::
316+
:header: Recurrent DQN
317+
:card_description: Use TorchRL to train recurrent policies
318+
:image: _static/img/rollout_recurrent.png
319+
:link: intermediate/dqn_with_rnn_tutorial.html
320+
:tags: Reinforcement-Learning
321+
315322
.. customcarditem::
316323
:header: Code a DDPG Loss
317324
:card_description: Use TorchRL to code a DDPG Loss
318325
:image: _static/img/half_cheetah.gif
319326
:link: advanced/coding_ddpg.html
320327
:tags: Reinforcement-Learning
321328

322-
323-
324329
.. Deploying PyTorch Models in Production
325330
326331

0 commit comments

Comments
 (0)