llama2.c icon indicating copy to clipboard operation
llama2.c copied to clipboard

can i train on cpu

Open pure-water opened this issue 1 year ago • 5 comments

pure-water avatar Dec 06 '23 02:12 pure-water

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

66RING avatar Dec 08 '23 13:12 66RING

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

pure-water avatar Dec 10 '23 05:12 pure-water

you solved it yet? @pure-water

YidaWeng avatar Jan 20 '24 07:01 YidaWeng

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

pure-water avatar Jan 21 '24 03:01 pure-water

OK, thanks anyway LOL

YidaWeng avatar Jan 22 '24 04:01 YidaWeng