dmlc / MXNet.jl

Calling mx.predict always prints an info message like: INFO: TempSpace: Total 0 MB allocated on CPU0, which can result in an overwhelming amount of printing when running lots of small predictions (I realize it's more efficient to do fewer predictions with larger batches, but that's not always possible). The relevant line is here:

MXNet.jl/src/model.jl

Line 155 in 9474d95

    
           info(string("TempSpace: ", split(dbg_str, ['\n'])[end-2]..., " on ", self.ctx[1]))

Would it be reasonable to add a verbosity keyword argument to predict()?

Suppress "TempSpace" info in predict