We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 29ae4c2 commit 747e543Copy full SHA for 747e543
examples/data_efficient_nopt/pretrain_basic.py
@@ -439,7 +439,7 @@ def train_one_epoch(self):
439
self.model.require_backward_grad_sync = (
440
1 + batch_idx
441
) % self.params.accum_grad == 0
442
- with amp.autocast(self.params.enable_amp, dtype=self.mp_type):
+ with amp.auto_cast(self.params.enable_amp, dtype=self.mp_type):
443
model_start = time.time()
444
445
output = self.model(inp_blur, mask)
0 commit comments