karpathy / llama2.c

Inference Llama 2 in one file of pure C

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

can i train on cpu

pure-water opened this issue · comments

can i train on cpu

yes. Just set device = "cpu" in train.py line 71.

i did not and there is a segmentation fault after moder initialized..... trying to get the hands dirty to debug it

you solved it yet? @pure-water

Not fiddling with it since got a gpu ....

OK, thanks anyway LOL