Support mixed precision #179

Open

Assignees

Labels

enhancementtodo

Milestone

opened

on Dec 25, 2025

Add another parameter in Trainer.train() to enable torch.amp.

to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Assignees

perctrix

Labels

enhancementtodo

Type

Projects

No projects

Milestone

1.2.x

Relationships

None yet

Development

No branches or pull requests