Dec. 28, 2023, 7:45 p.m. | /u/DataWizJesse

Machine Learning www.reddit.com

I have tried downloading the Tensorflow-Metal package for my multi-attention head transformer for disease risk prediction on sequence data. However when I use the GPU and I did validate it is being utilized, is it taking 3 times as long than the CPU. The batch sizes are set to 32 for the validation and testing. Any tips or ideas would be greatly appreciated :)

attention chip cpu data disease gpu head m3 pro machinelearning metal package prediction risk set tensorflow transformer

Data Engineer

@ Lemon.io | Remote: Europe, LATAM, Canada, UK, Asia, Oceania

Artificial Intelligence – Bioinformatic Expert

@ University of Texas Medical Branch | Galveston, TX

Lead Developer (AI)

@ Cere Network | San Francisco, US

Research Engineer

@ Allora Labs | Remote

Ecosystem Manager

@ Allora Labs | Remote

Founding AI Engineer, Agents

@ Occam AI | New York