pyblaze icon indicating copy to clipboard operation
pyblaze copied to clipboard

Potentially missing eval mode in `pyblaze.nn.engine.base.predict` method.

Open sharpenb opened this issue 4 years ago • 0 comments

https://github.com/borchero/pyblaze/blob/e0b35874ad9e56a6a016bfc6482ac90be6ffc1dc/pyblaze/nn/engine/base.py#L328

One might expect the predict method to set the model in evaluation mode.

sharpenb avatar Apr 29 '21 12:04 sharpenb