Adding appends to some of the pseudocode blocks (#8427)

This commit is contained in:
Caleb Robinson 2021-07-20 01:09:31 -07:00 committed by GitHub
parent 556879e5cf
commit 0ac9e2ae60
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions

View File

@ -213,6 +213,7 @@ Here's the pseudocode of what it does under the hood:
for batch in train_dataloader:
# forward
out = training_step(val_batch)
outs.append(out)
# clear gradients
optimizer.zero_grad()
@ -250,6 +251,7 @@ The matching pseudocode is:
for batch in train_dataloader:
# forward
out = training_step(val_batch)
outs.append(out)
# clear gradients
optimizer.zero_grad()