search

coqui-ai / Trainer

🐸 - A general purpose model trainer, as flexible as it gets
185 stars 109 forks source link

Remove AMP #136

Open mueller91 opened 9 months ago

mueller91 commented 9 months ago

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

erogol commented 9 months ago

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