eval_model_if_nn_module¶

tiatoolbox . models . architecture . utils . eval_model_if_nn_module

eval_model_if_nn_module(model)[source]¶

Run model.eval() if the model is a torch module.