search

joaopauloschuler / neural-api

CAI NEURAL API - Pascal based deep learning neural network API optimized for AVX, AVX2 and AVX512 instruction sets plus OpenCL capable devices including AMD, Intel and NVIDIA.
GNU Lesser General Public License v2.1
356 stars 195 forks source link

Enabling OpenCl #126

Open Dzandaa opened 8 months ago

Dzandaa commented 8 months ago

Hi,

Is it useful to enable OpenCL in both Network and Fit?

Thank you.

B->

mikerabat commented 8 months ago

Actually when you enabling it in the fitting class it is propagated to the networks (e.g. the TNNetDataParallelism object) when fit is called. So no.. you just need it to enable it in the fitting object.

Dzandaa commented 8 months ago

Thank you very much.

B->