Closed petebankhead closed 10 months ago
This makes memory management much easier, since there is no need for resources (NDList) to be accessible outside of the predict method. Appears to reduce memory leaks with TensorFlow + GPU.
NDList
predict
This makes memory management much easier, since there is no need for resources (
NDList
) to be accessible outside of thepredict
method. Appears to reduce memory leaks with TensorFlow + GPU.