Try pytorch inference mode #12

Open

Open

Try pytorch inference mode#12

torch.inference_mode is supposed to provide a small speedup.
https://pytorch.org/docs/stable/generated/torch.inference_mode.html

Metadata

Assignees

No one assigned

Labels

No labels

No labels

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests