fix typo in comment

This commit is contained in:
rasbt
2024-06-09 06:14:02 -05:00
parent e1adeb14f3
commit f0e4c99bc3
7 changed files with 9 additions and 9 deletions

View File

@@ -494,7 +494,7 @@ def train_classifier_simple(model, train_loader, val_loader, optimizer, device,
model.train() # Set model to training mode
for input_batch, target_batch in train_loader:
optimizer.zero_grad() # Reset loss gradients from previous epoch
optimizer.zero_grad() # Reset loss gradients from previous batch iteration
loss = calc_loss_batch(input_batch, target_batch, model, device)
loss.backward() # Calculate loss gradients
optimizer.step() # Update model weights using loss gradients