Closed jeffrey9909 closed 7 years ago
In general, you'll probably get better performance on your machine is you use train_gen.py
instead. You are getting an "out of memory" error because the default batch size is pretty big! (The repo developer runs a GTX1080). I have had my GTX970 work with a batch size of 300 iirc. So your command becomes python train_gen.py --batch_size 300 data/processed.h5 model.h5 --epochs 20
.
Oh, I see. Thanks for your help! I was thinking something like I need to change the code as the example I have mentioned. Thanks again! I will give it a try again.
I am new to ubuntu and this program, so this question may be very simple... This error message shows when I run the train.py
I think that this may be my GPU running out of memory. My graphic card is a GTX970 with 4GB memory. I am running the program according to the README, which is inputting this line
I have read a very similar problem as below
but I still cannot find a way to fix my problem. Can anyone please give me a hand? Thank you very much.