debug
parent
81b54c05d5
commit
8405882f11
|
@ -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')
|
||||
|
||||
|
|
Loading…
Reference in New Issue