pull/2526/head
gaotingquan 2023-05-29 08:19:20 +00:00 committed by Tingquan Gao
parent 81b54c05d5
commit 8405882f11
1 changed files with 1 additions and 1 deletions

View File

@ -579,7 +579,7 @@ class Engine(object):
optimizers=self.optimizer,
level=amp_level,
save_dtype='float32')
elif self.amp_eval:
elif amp_eval:
self.model = paddle.amp.decorate(
models=self.model, level=amp_level, save_dtype='float32')