You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
I run the "train_a_new_model_on_a_new_dataset_from_scratch" notebook on my M2 and no GPUs are detected, the script moves to using CPUs. Is it possible to configure Corenet to detect the M2 GPU?
The text was updated successfully, but these errors were encountered:
There's one way to do it quickly. If you are trying to use OpenELM, you will use LLM-Adapters and lm-evaluation-harness. I just searched and replaced ' "cpu" ' with' "mps" ' throughout the files. Also, I added the mps selection code in corenet file. (https://github.com/apple/corenet/pull/22/files). Worked well on M1 Pro.
I run the "train_a_new_model_on_a_new_dataset_from_scratch" notebook on my M2 and no GPUs are detected, the script moves to using CPUs. Is it possible to configure Corenet to detect the M2 GPU?
The text was updated successfully, but these errors were encountered: