Trainer icon indicating copy to clipboard operation
Trainer copied to clipboard

Remove AMP

Open mueller91 opened this issue 1 year ago • 1 comments

Describe the bug

Hi everybody,

Trainer is great, but still uses APEX, which is deprecated and tends to cause problems. Could you remove and/or replace with Nvidia/amp?

Best regrads

image

To Reproduce

Just run Trainer on an Nvidia GPU

Expected behavior

No response

Logs

No response

Environment

not relevant

Additional context

No response

mueller91 avatar Dec 12 '23 07:12 mueller91

But looks like all is put into torch natively, so yes there is no point keep it.

erogol avatar Dec 18 '23 07:12 erogol