Fixes #250 (#253)

2026-09-09 11:32:07 +08:00 · 2019-09-26 09:13:00 -04:00
parent b0a0a47a0b
commit 3337c0237b
1 changed files with 1 additions and 1 deletions
@@ -145,7 +145,7 @@ def training_step(self, batch, batch_nb):
    
    output = {
        'loss': loss, # required
-        'progress': {'training_loss': loss, 'batch_nb': batch_nb} # optional
+        'progress': {'training_loss': loss} # optional (MUST ALL BE TENSORS)
    }
    
    # return a dict