Support mixed precision #179

Open

Assignees

Labels

enhancementtodo

Milestone

opened

on Dec 25, 2025

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

Metadata

Assignees

perctrix

Labels

enhancementtodo

Type

Projects

No projects

Milestone

1.2.x

Relationships

None yet

Development

No branches or pull requests