Viewing a single comment thread. View all comments

1infiniteloop t1_iunkkee wrote

Tensorflow models seem to work well on GPU (M1 Pro 32 GB memory). With Pytorch I have had difficulty getting the model to train on the GPU. CPU training for Pytorch models has been very slow.

1