Skip to content

Commit 593837e

Browse files
committed
fix amp wrong call
1 parent 3ff6955 commit 593837e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

pytorch_lightning/trainer/training_loop.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -743,7 +743,8 @@ def call_optimizer_step(self, optimizer, opt_idx, batch_idx, split_batch):
743743
# when using 16-bit
744744
else:
745745
native_amp = self.use_amp and NATIVE_AMP_AVALAIBLE
746-
model.optimizer_step(self.current_epoch, batch_idx, optimizer, opt_idx, lambda_closure, native_amp)
746+
model.optimizer_step(self.current_epoch, batch_idx, optimizer, opt_idx, lambda_closure,
747+
using_native_amp=native_amp)
747748

748749
# in native 16-bit we need to update scaler after optimizer step
749750
if self.use_amp and NATIVE_AMP_AVALAIBLE:

0 commit comments

Comments
 (0)