🤔[question] can you provide me a example that use amp(mixed precision)

[ShiroZhang]

Describe your question

class AmpTrial(PyTorchTrial):
    pass

Checklist

• Did you search the docs for a solution?
• Did you search github issues to find if somebody asked this question before?

[ioga]

We don't require any of our own code for the amp. Simply follow the torch amp autocast docs, here's one example from our tests.

[ShiroZhang]

We don't require any of our own code for the amp. Simply follow the torch amp autocast docs, here's one example from our tests.

thank you