Put validation loss under amp_autocast

Browse Source

Secured the loss evaluation under the amp, avoiding function to operate on float16

...

This commit is contained in:

Lorenzo Baraldi 2022-12-09 12:03:23 +01:00

parent 7c4ed4d5a4

commit 3d6bc42aa1

2 changed files with 11 additions and 11 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Show Stats Download Patch File Download Diff File Expand all files Collapse all files