This repository has been archived by the owner on Mar 8, 2024. It is now read-only.

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

Open

sharpenb opened this issue Apr 29, 2021 · 0 comments

sharpenb commented Apr 29, 2021 •

edited

Loading

pyblaze/pyblaze/nn/engine/base.py

Line 328 in e0b3587

def predict(self, data, iterations=None, callbacks=None, gpu='auto', **kwargs):

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

The text was updated successfully, but these errors were encountered:

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.