ref: decoupled ddp spawn (#3746)

This commit is contained in:
William Falcon 2020-09-30 08:41:24 -04:00 committed by GitHub
parent 6b14ec5eee
commit d3ae26fcfc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions

View File

@ -393,7 +393,7 @@ If you want to calculate epoch-level metrics and log them, use the `.log` method
loss = F.cross_entropy(y_hat, y)
# logs metrics for each training_step, and the average across the epoch, to the progress bar and logger
result.log('train_loss', loss, on_step=True, on_epoch=True, prog_bar=True, logger=True)
self.log('train_loss', loss, on_step=True, on_epoch=True, prog_bar=True, logger=True)
return loss
The `.log` object automatically reduces the requested metrics across the full epoch.