Skip to content

Run in default mode when inference mode is disabled#146

Open
hakanardo wants to merge 1 commit into
triton-inference-server:mainfrom
Spiideo:grad
Open

Run in default mode when inference mode is disabled#146
hakanardo wants to merge 1 commit into
triton-inference-server:mainfrom
Spiideo:grad

Conversation

@hakanardo
Copy link
Copy Markdown

When INFERENCE_MODE is set to false, the model still runs in no_grad mode. Is that intentional? This prevents the serving of models that requires gradients at inference time, such as differential rendering for example. Can we instead use the default mode when INFERENCE_MODE=false (as is implemented in this pull-request) or would it be preferable to add additional parameters to enable the default mode?

Thanx!

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

Labels

None yet

Development

Successfully merging this pull request may close these issues.

1 participant