search

paganpasta / eqxvision

A Python package of computer vision models for the Equinox ecosystem.
https://eqxvision.readthedocs.io
MIT License
97 stars 11 forks source link

inference mode #32

Closed eforw closed 1 year ago

eforw commented 1 year ago

Is there an example how to switch to inference mode for evaluation?

paganpasta commented 1 year ago

Sure.

model = equinox.tree_inference(model, True)

eforw commented 1 year ago

Wow, thanks