Skip to content

Commit 747e543

Browse files
committed
feat(ppsci): support data_effient_nopt for training and test
1 parent 29ae4c2 commit 747e543

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/data_efficient_nopt/pretrain_basic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -439,7 +439,7 @@ def train_one_epoch(self):
439439
self.model.require_backward_grad_sync = (
440440
1 + batch_idx
441441
) % self.params.accum_grad == 0
442-
with amp.autocast(self.params.enable_amp, dtype=self.mp_type):
442+
with amp.auto_cast(self.params.enable_amp, dtype=self.mp_type):
443443
model_start = time.time()
444444

445445
output = self.model(inp_blur, mask)

0 commit comments

Comments
 (0)