ExplainableAI.jl icon indicating copy to clipboard operation
ExplainableAI.jl copied to clipboard
verified publisher icon Julia-XAI

Use multithreading in batch gradient computation

Open adrhill opened this issue 3 years ago • 0 comments

Add the Threads.@threads macro to the for-loop in gradients_wrt_batch: https://github.com/adrhill/ExplainableAI.jl/blob/f1b89ab9c784ff2a86de9997efa904580b2af6dd/src/gradient.jl#L5-L16

adrhill avatar May 24 '22 20:05 adrhill