Put validation loss under amp_autocast

Browse Source

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

pull/1586/head

Lorenzo Baraldi 2 years ago

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

Write Preview

Loading…

Cancel

Save