Adding appends to some of the pseudocode blocks (#8427)
This commit is contained in:
parent
556879e5cf
commit
0ac9e2ae60
|
@ -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()
|
||||
|
|
Loading…
Reference in New Issue